Searched refs:nchwShape (Results 1 – 1 of 1) sorted by relevance
117 inline bool convertNchwToNhwc(const T* nchw, const Shape& nchwShape, std::vector<T>* nhwc, in convertNchwToNhwc() argument119 NN_RET_CHECK_EQ(getNumberOfDimensions(nchwShape), 4) in convertNchwToNhwc()121 *nhwcShape = nchwShape; in convertNchwToNhwc()122 const auto& fromDim = nchwShape.dimensions; in convertNchwToNhwc()124 nhwc->resize(getNumberOfElements(nchwShape)); in convertNchwToNhwc()