Home
last modified time | relevance | path

Searched refs:batchesOutData (Results 1 – 1 of 1) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DGenerateProposals.cpp650 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()
1118 batchesOutData[i] = batchesOut[i]; in generateProposalsFloat32()
1166 batchesOutData[i] = batchesOut[i]; in generateProposalsFloat16()
[all …]