Searched refs:heatmap_nhwc (Results 1 – 1 of 1) sorted by relevance
164 std::vector<float> heatmap_nhwc; in heatmapMaxKeypointFloat32() local167 NN_RET_CHECK(convertNchwToNhwc(heatmap, heatmapShape, &heatmap_nhwc, &heatmapShape_nhwc)); in heatmapMaxKeypointFloat32()169 const float* heatmap_tmp = layout ? heatmap_nhwc.data() : heatmap; in heatmapMaxKeypointFloat32()