Lines Matching refs:batchesData

62                                  const int32_t* batchesData, const Shape& batchesShape,  in bboxTransformFloat32()  argument
76 uint32_t batchIndex = batchesData[roiIndex]; in bboxTransformFloat32()
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()
490 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitFloat32Compute() argument
514 if (batchesData[i] == ind) { in boxWithNmsLimitFloat32Compute()
517 ind = batchesData[i]; in boxWithNmsLimitFloat32Compute()
621 const Shape& roiShape, const int32_t* batchesData, 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
658 scores_float32.data(), scoresShape, roi_float32.data(), roiShape, batchesData, in boxWithNmsLimitFloat16()
667 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitQuant() argument
680 scores_float32.data(), scoresShape, roi_float32.data(), roiShape, batchesData, in boxWithNmsLimitQuant()
689 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitQuant() argument
703 scores_float32.data(), scoresShape, roi_float32.data(), roiShape, batchesData, in boxWithNmsLimitQuant()