Lines Matching refs:batchesOutData
626 int32_t* classesOutData, Shape classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitFloat32() argument
650 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitFloat16() argument
672 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitQuant() argument
694 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitQuant() argument
967 std::vector<int32_t>* batchesOutData) { in generateProposalsNhwcFloat32Compute() argument
981 batchesOutData->clear(); in generateProposalsNhwcFloat32Compute()
1052 batchesOutData->push_back(b); in generateProposalsNhwcFloat32Compute()
1069 std::vector<int32_t>* batchesOutData) { in generateProposalsFloat32Compute() argument
1077 scoresOutData, roiOutData, batchesOutData); in generateProposalsFloat32Compute()
1116 int32_t* batchesOutData = context->getOutputBuffer<int32_t>(kOutputBatchesTensor); in generateProposalsFloat32() local
1118 batchesOutData[i] = batchesOut[i]; in generateProposalsFloat32()
1164 int32_t* batchesOutData = context->getOutputBuffer<int32_t>(kOutputBatchesTensor); in generateProposalsFloat16() local
1166 batchesOutData[i] = batchesOut[i]; in generateProposalsFloat16()
1217 int32_t* batchesOutData = context->getOutputBuffer<int32_t>(kOutputBatchesTensor); in generateProposalsQuant() local
1219 batchesOutData[i] = batchesOut[i]; in generateProposalsQuant()