88 inline bool heatmapMaxKeypointFloat32Nhwc(const float* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointFloat32Nhwc()
89                                           const float* boxes, const Shape& boxesShape,  in heatmapMaxKeypointFloat32Nhwc()
90                                           float* outputScoreData, const Shape& outputScoreShape,  in heatmapMaxKeypointFloat32Nhwc()
92                                           const Shape& outputKeypointShape, float fpAtol,  in heatmapMaxKeypointFloat32Nhwc()
167 inline bool heatmapMaxKeypointFloat32(const float* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointFloat32()
168                                       const float* boxes, const Shape& boxesShape, bool layout,  in heatmapMaxKeypointFloat32()
169                                       float* outputScoreData, const Shape& outputScoreShape,  in heatmapMaxKeypointFloat32()
170                                       float* outputKeypointData, const Shape& outputKeypointShape,  in heatmapMaxKeypointFloat32()
173     Shape heatmapShape_nhwc;  in heatmapMaxKeypointFloat32()
178     const Shape& heatmapShape_tmp = layout ? heatmapShape_nhwc : heatmapShape;  in heatmapMaxKeypointFloat32()
184 inline bool heatmapMaxKeypointQuant(const uint8_t* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointQuant()
185                                     const uint16_t* boxes, const Shape& boxesShape, bool layout,  in heatmapMaxKeypointQuant()
186                                     uint8_t* outputScoreData, const Shape& outputScoreShape,  in heatmapMaxKeypointQuant()
187                                     uint16_t* outputKeypointData, const Shape& outputKeypointShape,  in heatmapMaxKeypointQuant()
206 inline bool heatmapMaxKeypointQuant(const int8_t* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointQuant()
207                                     const uint16_t* boxes, const Shape& boxesShape, bool layout,  in heatmapMaxKeypointQuant()
208                                     int8_t* outputScoreData, const Shape& outputScoreShape,  in heatmapMaxKeypointQuant()
209                                     uint16_t* outputKeypointData, const Shape& outputKeypointShape,  in heatmapMaxKeypointQuant()
261     Shape heatmapShape = context->getInputShape(kHeatmapTensor);  in prepare()
262     Shape boxesShape = context->getInputShape(kBoxesTensor);  in prepare()
281     Shape outputScore = context->getOutputShape(kOutputScoreTensor);  in prepare()
286     Shape outputKeypoint = context->getOutputShape(kOutputKeypointTensor);  in prepare()