Lines Matching refs:heatmap
89 inline bool heatmapMaxKeypointFloat32Nhwc(const float* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointFloat32Nhwc() argument
102 const float* heatmapBase = heatmap; in heatmapMaxKeypointFloat32Nhwc()
168 inline bool heatmapMaxKeypointFloat32(const float* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointFloat32() argument
176 NN_RET_CHECK(convertNchwToNhwc(heatmap, heatmapShape, &heatmap_nhwc, &heatmapShape_nhwc)); in heatmapMaxKeypointFloat32()
178 const float* heatmap_tmp = layout ? heatmap_nhwc.data() : heatmap; in heatmapMaxKeypointFloat32()
185 inline bool heatmapMaxKeypointQuant(const uint8_t* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointQuant() argument
191 convertQuantToFloat32(heatmap, heatmapShape.scale, heatmapShape.offset, &heatmap_float32); in heatmapMaxKeypointQuant()
207 inline bool heatmapMaxKeypointQuant(const int8_t* heatmap, const Shape& heatmapShape, in heatmapMaxKeypointQuant() argument
213 convertQuantToFloat32(heatmap, heatmapShape.scale, heatmapShape.offset, &heatmap_float32); in heatmapMaxKeypointQuant()
305 const auto heatmap = context->getInputBuffer<_Float16>(kHeatmapTensor); in execute() local
314 convertFloat16ToFloat32(heatmap, &heatmap_float32); in execute()