Searched refs:outputKeypointShape (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | HeatmapMaxKeypoint.cpp | 162 float* outputKeypointData, const Shape& outputKeypointShape, in heatmapMaxKeypointFloat32() argument
173 outputKeypointShape, fpAtol, fpRtol); in heatmapMaxKeypointFloat32()
179 uint16_t* outputKeypointData, const Shape& outputKeypointShape, in heatmapMaxKeypointQuant() argument
186 std::vector<float> outputKeypoint_float32(getNumberOfElements(outputKeypointShape)); in heatmapMaxKeypointQuant()
190 outputKeypointShape, fpAtol, fpRtol)); in heatmapMaxKeypointQuant()
193 convertFloat32ToQuant(outputKeypoint_float32, outputKeypointShape.scale, in heatmapMaxKeypointQuant()
194 outputKeypointShape.offset, outputKeypointData); in heatmapMaxKeypointQuant()
201 uint16_t* outputKeypointData, const Shape& outputKeypointShape, in heatmapMaxKeypointQuant() argument
208 std::vector<float> outputKeypoint_float32(getNumberOfElements(outputKeypointShape)); in heatmapMaxKeypointQuant()
212 outputKeypointShape, fpAtol, fpRtol)); in heatmapMaxKeypointQuant()
[all …]
|