Home
last modified time | relevance | path

Searched refs:heatmapShape_nhwc (Results 1 – 1 of 1) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DHeatmapMaxKeypoint.cpp165 Shape heatmapShape_nhwc; in heatmapMaxKeypointFloat32() local
167 NN_RET_CHECK(convertNchwToNhwc(heatmap, heatmapShape, &heatmap_nhwc, &heatmapShape_nhwc)); in heatmapMaxKeypointFloat32()
170 const Shape& heatmapShape_tmp = layout ? heatmapShape_nhwc : heatmapShape; in heatmapMaxKeypointFloat32()