Searched refs:batchesShape (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | GenerateProposals.cpp | 123 bboxDeltasShape, batchesData, batchesShape, in bboxTransformFloat16()
145 bboxDeltasShape, batchesData, batchesShape, in bboxTransformQuant()
167 bboxDeltasShape, batchesData, batchesShape, in bboxTransformQuant()
233 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare() local
239 NN_RET_CHECK_EQ(getNumberOfDimensions(batchesShape), 1); in prepare()
252 NN_RET_CHECK_EQ(getSizeOfDimension(batchesShape, 0), numRois); in prepare()
622 const Shape& batchesShape, float scoreThreshold, in boxWithNmsLimitFloat32() argument
762 Shape batchesShape = context->getInputShape(kBatchesTensor); in prepare() local
770 NN_RET_CHECK(getNumberOfDimensions(batchesShape) == 1); in prepare()
778 NN_RET_CHECK(getSizeOfDimension(batchesShape, 0) == numRois); in prepare()
[all …]
|