Searched refs:convertNchwToNhwc (Results 1 – 2 of 2) sorted by relevance
118 inline bool convertNchwToNhwc(const T* nchw, const Shape& nchwShape, std::vector<T>* nhwc, in convertNchwToNhwc() function166 return convertNchwToNhwc(mDataOriginal, shape, &mDataNhwc, &mShape); in initialize()
176 NN_RET_CHECK(convertNchwToNhwc(heatmap, heatmapShape, &heatmap_nhwc, &heatmapShape_nhwc)); in heatmapMaxKeypointFloat32()