Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DRoiPooling.cpp38 constexpr uint32_t kRoiTensor = 1; variable
232 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
281 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
282 context->getInputShape(kRoiTensor), in execute()
293 context->getInputBuffer<float>(kRoiTensor), in execute()
294 context->getInputShape(kRoiTensor), in execute()
305 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
306 context->getInputShape(kRoiTensor), in execute()
317 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
318 context->getInputShape(kRoiTensor), in execute()
DRoiAlign.cpp39 constexpr uint32_t kRoiTensor = 1; variable
387 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
444 if (getNumberOfElements(context->getInputShape(kRoiTensor)) == 0) return true; in execute()
449 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
450 context->getInputShape(kRoiTensor), in execute()
463 context->getInputBuffer<float>(kRoiTensor), in execute()
464 context->getInputShape(kRoiTensor), in execute()
477 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
478 context->getInputShape(kRoiTensor), in execute()
491 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
[all …]
DGenerateProposals.cpp195 constexpr uint32_t kRoiTensor = 0; variable
207 auto inputType = context->getInputType(kRoiTensor); in validate()
230 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
276 switch (context->getInputType(kRoiTensor)) { in execute()
278 return bboxTransformFloat16(context->getInputBuffer<_Float16>(kRoiTensor), in execute()
279 context->getInputShape(kRoiTensor), in execute()
290 return bboxTransformFloat32(context->getInputBuffer<float>(kRoiTensor), in execute()
291 context->getInputShape(kRoiTensor), in execute()
303 return bboxTransformQuant(context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
304 context->getInputShape(kRoiTensor), in execute()
[all …]