Lines Matching refs:kRoiTensor
197 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()
279 return bboxTransformFloat16(context->getInputBuffer<_Float16>(kRoiTensor), in execute()
280 context->getInputShape(kRoiTensor), in execute()
291 return bboxTransformFloat32(context->getInputBuffer<float>(kRoiTensor), in execute()
292 context->getInputShape(kRoiTensor), in execute()
304 return bboxTransformQuant(context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
305 context->getInputShape(kRoiTensor), in execute()
315 return bboxTransformQuant(context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
316 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()
823 context->getInputShape(kRoiTensor), in execute()
844 context->getInputBuffer<float>(kRoiTensor), in execute()
845 context->getInputShape(kRoiTensor), in execute()
866 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
867 context->getInputShape(kRoiTensor), in execute()
888 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
889 context->getInputShape(kRoiTensor), in execute()