Lines Matching refs:kRoiTensor
199 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()
547 const T_Roi* roiBase = context->getInputBuffer<T_Roi>(kRoiTensor); in boxWithNmsLimitWriteOutput()
668 Shape roiShape = context->getInputShape(kRoiTensor); in prepare()
729 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
730 context->getInputShape(kRoiTensor), in execute()
751 context->getInputBuffer<float>(kRoiTensor), in execute()
752 context->getInputShape(kRoiTensor), in execute()
773 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
774 context->getInputShape(kRoiTensor), in execute()
795 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
796 context->getInputShape(kRoiTensor), in execute()