Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DRoiPooling.cpp40 constexpr uint32_t kRoiTensor = 1; variable
233 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
282 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
283 context->getInputShape(kRoiTensor), in execute()
294 context->getInputBuffer<float>(kRoiTensor), in execute()
295 context->getInputShape(kRoiTensor), in execute()
306 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
307 context->getInputShape(kRoiTensor), in execute()
318 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
319 context->getInputShape(kRoiTensor), in execute()
H A DRoiAlign.cpp42 constexpr uint32_t kRoiTensor = 1; variable
390 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
447 if (getNumberOfElements(context->getInputShape(kRoiTensor)) == 0) return true; in execute()
452 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
453 context->getInputShape(kRoiTensor), in execute()
466 context->getInputBuffer<float>(kRoiTensor), in execute()
467 context->getInputShape(kRoiTensor), in execute()
480 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
481 context->getInputShape(kRoiTensor), in execute()
494 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
[all …]
H A DGenerateProposals.cpp197 constexpr uint32_t kRoiTensor = 0; variable
209 auto inputType = context->getInputType(kRoiTensor); in validate()
231 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
277 switch (context->getInputType(kRoiTensor)) { in execute()
280 context->getInputShape(kRoiTensor), in execute()
292 context->getInputShape(kRoiTensor), in execute()
341 constexpr uint32_t kRoiTensor = 1; variable
597 const T_Roi* roiBase = context->getInputBuffer<T_Roi>(kRoiTensor); in boxWithNmsLimitWriteOutput()
761 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
822 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
[all …]