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