Home
last modified time | relevance | path

Searched refs:kRoiTensor (Results 1 – 7 of 7) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DRoiPooling.cpp181 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
230 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
231 context->getInputShape(kRoiTensor), in execute()
242 context->getInputBuffer<float>(kRoiTensor), in execute()
243 context->getInputShape(kRoiTensor), in execute()
254 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
255 context->getInputShape(kRoiTensor), in execute()
266 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
267 context->getInputShape(kRoiTensor), in execute()
DRoiAlign.cpp335 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
392 if (getNumberOfElements(context->getInputShape(kRoiTensor)) == 0) return true; in execute()
397 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
398 context->getInputShape(kRoiTensor), in execute()
411 context->getInputBuffer<float>(kRoiTensor), in execute()
412 context->getInputShape(kRoiTensor), in execute()
425 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
426 context->getInputShape(kRoiTensor), in execute()
439 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
440 context->getInputShape(kRoiTensor), in execute()
DGenerateProposals.cpp199 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
245 switch (context->getInputType(kRoiTensor)) { in execute()
247 return bboxTransformFloat16(context->getInputBuffer<_Float16>(kRoiTensor), in execute()
248 context->getInputShape(kRoiTensor), in execute()
259 return bboxTransformFloat32(context->getInputBuffer<float>(kRoiTensor), in execute()
260 context->getInputShape(kRoiTensor), in execute()
272 return bboxTransformQuant(context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
273 context->getInputShape(kRoiTensor), in execute()
283 return bboxTransformQuant(context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
284 context->getInputShape(kRoiTensor), in execute()
[all …]
/packages/modules/NeuralNetworks/common/types/operations/include/
DGenerateProposals.h29 constexpr uint32_t kRoiTensor = 0; variable
45 constexpr uint32_t kRoiTensor = 1; variable
DRoiPooling.h28 constexpr uint32_t kRoiTensor = 1; variable
DRoiAlign.h28 constexpr uint32_t kRoiTensor = 1; variable
/packages/modules/NeuralNetworks/common/types/operations/src/
DGenerateProposals.cpp32 auto inputType = context->getInputType(kRoiTensor); in validate()