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