89 inline bool heatmapMaxKeypointFloat32Nhwc(const float* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointFloat32Nhwc()
90                                           const float* boxes, const Shape& boxesShape,  in heatmapMaxKeypointFloat32Nhwc()
91                                           float* outputScoreData, const Shape& outputScoreShape,  in heatmapMaxKeypointFloat32Nhwc()
93                                           const Shape& outputKeypointShape, float fpAtol,  in heatmapMaxKeypointFloat32Nhwc()
168 inline bool heatmapMaxKeypointFloat32(const float* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointFloat32()
169                                       const float* boxes, const Shape& boxesShape, bool layout,  in heatmapMaxKeypointFloat32()
170                                       float* outputScoreData, const Shape& outputScoreShape,  in heatmapMaxKeypointFloat32()
171                                       float* outputKeypointData, const Shape& outputKeypointShape,  in heatmapMaxKeypointFloat32()
174     Shape heatmapShape_nhwc;  in heatmapMaxKeypointFloat32()
179     const Shape& heatmapShape_tmp = layout ? heatmapShape_nhwc : heatmapShape;  in heatmapMaxKeypointFloat32()
185 inline bool heatmapMaxKeypointQuant(const uint8_t* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointQuant()
186                                     const uint16_t* boxes, const Shape& boxesShape, bool layout,  in heatmapMaxKeypointQuant()
187                                     uint8_t* outputScoreData, const Shape& outputScoreShape,  in heatmapMaxKeypointQuant()
188                                     uint16_t* outputKeypointData, const Shape& outputKeypointShape,  in heatmapMaxKeypointQuant()
207 inline bool heatmapMaxKeypointQuant(const int8_t* heatmap, const Shape& heatmapShape,  in heatmapMaxKeypointQuant()
208                                     const uint16_t* boxes, const Shape& boxesShape, bool layout,  in heatmapMaxKeypointQuant()
209                                     int8_t* outputScoreData, const Shape& outputScoreShape,  in heatmapMaxKeypointQuant()
210                                     uint16_t* outputKeypointData, const Shape& outputKeypointShape,  in heatmapMaxKeypointQuant()
263     Shape heatmapShape = context->getInputShape(kHeatmapTensor);  in prepare()
264     Shape boxesShape = context->getInputShape(kBoxesTensor);  in prepare()
283     Shape outputScore = context->getOutputShape(kOutputScoreTensor);  in prepare()
288     Shape outputKeypoint = context->getOutputShape(kOutputKeypointTensor);  in prepare()