Searched refs:outputScoreData (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | HeatmapMaxKeypoint.cpp | 91 float* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointFloat32Nhwc() argument
104 float* outputScoreBase = outputScoreData; in heatmapMaxKeypointFloat32Nhwc()
170 float* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointFloat32() argument
181 outputScoreData, outputScoreShape, outputKeypointData, in heatmapMaxKeypointFloat32()
187 uint8_t* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointQuant() argument
201 outputScoreData); in heatmapMaxKeypointQuant()
209 int8_t* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointQuant() argument
223 outputScoreData); in heatmapMaxKeypointQuant()
309 auto outputScoreData = context->getOutputBuffer<_Float16>(kOutputScoreTensor); in execute() local
323 convertFloat32ToFloat16(outputScore_float32, outputScoreData); in execute()
|