Searched refs:outputKeypointShape (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | HeatmapMaxKeypoint.cpp | 93 const Shape& outputKeypointShape, float fpAtol, in heatmapMaxKeypointFloat32Nhwc() argument
182 outputKeypointShape, fpAtol, fpRtol); in heatmapMaxKeypointFloat32()
195 std::vector<float> outputKeypoint_float32(getNumberOfElements(outputKeypointShape)); in heatmapMaxKeypointQuant()
199 outputKeypointShape, fpAtol, fpRtol)); in heatmapMaxKeypointQuant()
202 convertFloat32ToQuant(outputKeypoint_float32, outputKeypointShape.scale, in heatmapMaxKeypointQuant()
203 outputKeypointShape.offset, outputKeypointData); in heatmapMaxKeypointQuant()
217 std::vector<float> outputKeypoint_float32(getNumberOfElements(outputKeypointShape)); in heatmapMaxKeypointQuant()
221 outputKeypointShape, fpAtol, fpRtol)); in heatmapMaxKeypointQuant()
224 convertFloat32ToQuant(outputKeypoint_float32, outputKeypointShape.scale, in heatmapMaxKeypointQuant()
225 outputKeypointShape.offset, outputKeypointData); in heatmapMaxKeypointQuant()
[all …]
|