Searched refs:nhwcShape (Results 1 – 1 of 1) sorted by relevance
100 Shape* nhwcShape) { in convertNchwToNhwc() argument103 *nhwcShape = nchwShape; in convertNchwToNhwc()105 nhwcShape->dimensions = {fromDim[0], fromDim[2], fromDim[3], fromDim[1]}; in convertNchwToNhwc()121 inline bool convertNhwcToNchw(const std::vector<T>& nhwc, const Shape& nhwcShape, T* nchw) { in convertNhwcToNchw() argument122 NN_RET_CHECK_EQ(getNumberOfDimensions(nhwcShape), 4) in convertNhwcToNchw()124 const auto& fromDim = nhwcShape.dimensions; in convertNhwcToNchw()