Searched defs:batchesShape (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | GenerateProposals.cpp | 114 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformFloat16()
134 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformQuant()
156 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformQuant()
200 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare() local
571 const Shape& batchesShape, float scoreThreshold, in boxWithNmsLimitFloat32()
594 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitFloat16()
618 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitQuant()
641 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitQuant()
668 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare() local
|