Searched refs:batchesData (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() argument
76 uint32_t batchIndex = batchesData[roiIndex]; in bboxTransformFloat32()
123 bboxDeltasShape, batchesData, batchesShape, in bboxTransformFloat16()
132 const int32_t* batchesData, const Shape& batchesShape, in bboxTransformQuant() argument
145 bboxDeltasShape, batchesData, batchesShape, in bboxTransformQuant()
167 bboxDeltasShape, batchesData, batchesShape, in bboxTransformQuant()
514 if (batchesData[i] == ind) { in boxWithNmsLimitFloat32Compute()
517 ind = batchesData[i]; in boxWithNmsLimitFloat32Compute()
621 const Shape& roiShape, const int32_t* batchesData, in boxWithNmsLimitFloat32() argument
667 const int32_t* batchesData, const Shape& batchesShape, in boxWithNmsLimitQuant() argument
[all …]
|