Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DGenerateProposals.cpp626 int32_t* classesOutData, Shape classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitFloat32()
650 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitFloat16()
672 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitQuant()
694 const Shape& classesOutShape, int32_t* batchesOutData, in boxWithNmsLimitQuant()
967 std::vector<int32_t>* batchesOutData) { in generateProposalsNhwcFloat32Compute()
1069 std::vector<int32_t>* batchesOutData) { in generateProposalsFloat32Compute()
1116 int32_t* batchesOutData = context->getOutputBuffer<int32_t>(kOutputBatchesTensor); in generateProposalsFloat32() local
1164 int32_t* batchesOutData = context->getOutputBuffer<int32_t>(kOutputBatchesTensor); in generateProposalsFloat16() local
1217 int32_t* batchesOutData = context->getOutputBuffer<int32_t>(kOutputBatchesTensor); in generateProposalsQuant() local