Lines Matching refs:getOutputShape
198 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare()
236 if (getNumberOfElements(context->getOutputShape(kOutputTensor)) == 0) return true; in execute()
248 context->getOutputShape(kOutputTensor)); in execute()
260 context->getOutputShape(kOutputTensor)); in execute()
272 context->getOutputShape(kOutputTensor)); in execute()
522 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()
526 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in boxWithNmsLimitWriteOutput()
530 Shape classesOutShape = context->getOutputShape(kOutputClassTensor); in boxWithNmsLimitWriteOutput()
534 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in boxWithNmsLimitWriteOutput()
676 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
677 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
678 Shape outputClassShape = context->getOutputShape(kOutputClassTensor); in prepare()
679 Shape outputBatchSplitShape = context->getOutputShape(kOutputBatchesTensor); in prepare()
741 context->getOutputShape(kOutputScoreTensor), in execute()
743 context->getOutputShape(kOutputRoiTensor), in execute()
745 context->getOutputShape(kOutputClassTensor), in execute()
747 context->getOutputShape(kOutputBatchesTensor), context); in execute()
763 context->getOutputShape(kOutputScoreTensor), in execute()
765 context->getOutputShape(kOutputRoiTensor), in execute()
767 context->getOutputShape(kOutputClassTensor), in execute()
769 context->getOutputShape(kOutputBatchesTensor), context); in execute()
785 context->getOutputShape(kOutputScoreTensor), in execute()
787 context->getOutputShape(kOutputRoiTensor), in execute()
789 context->getOutputShape(kOutputClassTensor), in execute()
791 context->getOutputShape(kOutputBatchesTensor), context); in execute()
981 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsFloat32()
984 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsFloat32()
987 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in generateProposalsFloat32()
1033 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsFloat16()
1036 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsFloat16()
1039 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in generateProposalsFloat16()
1083 Shape scoresOutShape = context->getOutputShape(kOutputScoreTensor); in generateProposalsQuant()
1086 Shape roiOutShape = context->getOutputShape(kOutputRoiTensor); in generateProposalsQuant()
1089 Shape batchesOutShape = context->getOutputShape(kOutputBatchesTensor); in generateProposalsQuant()
1170 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
1171 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
1172 Shape outputBatchSplitShape = context->getOutputShape(kOutputBatchesTensor); in prepare()
1477 Shape outputScoreShape = context->getOutputShape(kOutputScoreTensor); in prepare()
1478 Shape outputRoiShape = context->getOutputShape(kOutputRoiTensor); in prepare()
1479 Shape outputClassShape = context->getOutputShape(kOutputClassTensor); in prepare()
1480 Shape outputDetectionShape = context->getOutputShape(kOutputDetectionTensor); in prepare()
1569 context->getOutputShape(kOutputScoreTensor), in execute()
1571 context->getOutputShape(kOutputRoiTensor), in execute()
1573 context->getOutputShape(kOutputClassTensor), in execute()
1575 context->getOutputShape(kOutputDetectionTensor)); in execute()
1597 context->getOutputShape(kOutputScoreTensor), in execute()
1599 context->getOutputShape(kOutputRoiTensor), in execute()
1601 context->getOutputShape(kOutputClassTensor), in execute()
1603 context->getOutputShape(kOutputDetectionTensor)); in execute()