Lines Matching refs:kOutputRoiTensor

533     Shape roiOutShape = context->getOutputShape(kOutputRoiTensor);  in boxWithNmsLimitWriteOutput()
535 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in boxWithNmsLimitWriteOutput()
550 T_Roi* roiOutPtr = context->getOutputBuffer<T_Roi>(kOutputRoiTensor); in boxWithNmsLimitWriteOutput()
671 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
708 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
741 context->getOutputBuffer<_Float16>(kOutputRoiTensor), in execute()
742 context->getOutputShape(kOutputRoiTensor), in execute()
763 context->getOutputBuffer<float>(kOutputRoiTensor), in execute()
764 context->getOutputShape(kOutputRoiTensor), in execute()
785 context->getOutputBuffer<uint16_t>(kOutputRoiTensor), in execute()
786 context->getOutputShape(kOutputRoiTensor), in execute()
807 context->getOutputBuffer<uint16_t>(kOutputRoiTensor), in execute()
808 context->getOutputShape(kOutputRoiTensor), in execute()
987 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsFloat32()
989 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in generateProposalsFloat32()
999 float* roiOutData = context->getOutputBuffer<float>(kOutputRoiTensor); in generateProposalsFloat32()
1039 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsFloat16()
1041 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in generateProposalsFloat16()
1049 _Float16* roiOutData = context->getOutputBuffer<_Float16>(kOutputRoiTensor); in generateProposalsFloat16()
1091 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsQuant()
1093 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in generateProposalsQuant()
1102 uint16_t* roiOutData = context->getOutputBuffer<uint16_t>(kOutputRoiTensor); in generateProposalsQuant()
1120 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
1160 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
1394 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
1450 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
1486 context->getOutputBuffer<_Float16>(kOutputRoiTensor), in execute()
1487 context->getOutputShape(kOutputRoiTensor), in execute()
1514 context->getOutputBuffer<float>(kOutputRoiTensor), in execute()
1515 context->getOutputShape(kOutputRoiTensor), in execute()