Lines Matching refs:batchesShape
62 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformFloat32() argument
112 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformFloat16() argument
123 bboxDeltasShape, batchesData, batchesShape, in bboxTransformFloat16()
132 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformQuant() argument
145 bboxDeltasShape, batchesData, batchesShape, in bboxTransformQuant()
154 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformQuant() argument
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()
490 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitFloat32Compute() argument
622 const Shape& batchesShape, float scoreThreshold, in boxWithNmsLimitFloat32() argument
635 scores_float32.data(), scoresShape, roiData, roiShape, batchesData, batchesShape, in boxWithNmsLimitFloat32()
644 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitFloat16() argument
659 batchesShape, scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, in boxWithNmsLimitFloat16()
667 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitQuant() argument
681 batchesShape, scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, in boxWithNmsLimitQuant()
689 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitQuant() argument
704 batchesShape, scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, in boxWithNmsLimitQuant()
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()
807 outputBatchSplitShape.type = batchesShape.type; in prepare()