Searched refs:outputScoreData (Results 1 – 1 of 1) sorted by relevance
/packages/modules/NeuralNetworks/common/cpu_operations/ |
D | HeatmapMaxKeypoint.cpp | 82 float* outputScoreData, const Shape& /*outputScoreShape*/, in heatmapMaxKeypointFloat32Nhwc() argument 95 float* outputScoreBase = outputScoreData; in heatmapMaxKeypointFloat32Nhwc() 161 float* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointFloat32() argument 172 outputScoreData, outputScoreShape, outputKeypointData, in heatmapMaxKeypointFloat32() 178 uint8_t* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointQuant() argument 192 outputScoreData); in heatmapMaxKeypointQuant() 200 int8_t* outputScoreData, const Shape& outputScoreShape, in heatmapMaxKeypointQuant() argument 214 outputScoreData); in heatmapMaxKeypointQuant() 270 auto outputScoreData = context->getOutputBuffer<_Float16>(kOutputScoreTensor); in execute() local 284 convertFloat32ToFloat16(outputScore_float32, outputScoreData); in execute()
|