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