Searched defs:batchesShape (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | GenerateProposals.cpp | 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()
[all …]
|