Searched refs:classOutShape (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | GenerateProposals.cpp | 1458 const Shape& classOutShape, int32_t* detectionOutData, const Shape& detectionOutShape) { in detectionPostprocessFloat32() argument
1468 memset(classOutData, 0, getNumberOfElements(classOutShape) * sizeof(int32_t)); in detectionPostprocessFloat32()
1563 const Shape& classOutShape, int32_t* detectionOutData, const Shape& detectionOutShape) { in detectionPostprocessFloat16() argument
1577 outputRoi_float32.data(), roiOutShape, classOutData, classOutShape, detectionOutData, in detectionPostprocessFloat16()
|