GenerateProposals.cpp - OpenGrok cross reference for /aosp12/packages/modules/NeuralNetworks/common/operations/GenerateProposals.cpp

Lines Matching defs:batchesShape

62                                  const int32_t* batchesData, const Shape& batchesShape,  in bboxTransformFloat32()
112                                  const int32_t* batchesData, const Shape& batchesShape,  in bboxTransformFloat16()
132                                const int32_t* batchesData, const Shape& batchesShape,  in bboxTransformQuant()
154                                const int32_t* batchesData, const Shape& batchesShape,  in bboxTransformQuant()
233     Shape batchesShape = context->getInputShape(kBatchesTensor);  in prepare()  local
490                                    const int32_t* batchesData, const Shape& batchesShape,  in boxWithNmsLimitFloat32Compute()
622                             const Shape& batchesShape, float scoreThreshold,  in boxWithNmsLimitFloat32()
644                             const int32_t* batchesData, const Shape& batchesShape,  in boxWithNmsLimitFloat16()
667                           const int32_t* batchesData, const Shape& batchesShape,  in boxWithNmsLimitQuant()
689                           const int32_t* batchesData, const Shape& batchesShape,  in boxWithNmsLimitQuant()
762     Shape batchesShape = context->getInputShape(kBatchesTensor);  in prepare()  local