• Home
  • Raw
  • Download

Lines Matching refs:setOutputShape

229     NN_RET_CHECK(context->setOutputShape(kOutputTensor, outputShape));  in prepare()
524 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in boxWithNmsLimitWriteOutput()
528 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in boxWithNmsLimitWriteOutput()
532 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, classesOutShape)); in boxWithNmsLimitWriteOutput()
536 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, batchesOutShape)); in boxWithNmsLimitWriteOutput()
703 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScoreShape)); in prepare()
709 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
713 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, outputClassShape)); in prepare()
717 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, outputBatchSplitShape)); in prepare()
983 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in generateProposalsFloat32()
986 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in generateProposalsFloat32()
989 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, batchesOutShape)); in generateProposalsFloat32()
1035 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in generateProposalsFloat16()
1038 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in generateProposalsFloat16()
1041 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, batchesOutShape)); in generateProposalsFloat16()
1085 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in generateProposalsQuant()
1088 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in generateProposalsQuant()
1091 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, batchesOutShape)); in generateProposalsQuant()
1204 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScoreShape)); in prepare()
1211 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
1214 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, outputBatchSplitShape)); in prepare()
1530 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScoreShape)); in prepare()
1534 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
1538 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, outputClassShape)); in prepare()
1542 NN_RET_CHECK(context->setOutputShape(kOutputDetectionTensor, outputDetectionShape)); in prepare()