Lines Matching refs:fromDim
104 const auto& fromDim = nchwShape.dimensions; in convertNchwToNhwc() local
105 nhwcShape->dimensions = {fromDim[0], fromDim[2], fromDim[3], fromDim[1]}; in convertNchwToNhwc()
108 uint32_t spatialSize = fromDim[2] * fromDim[3]; in convertNchwToNhwc()
109 for (uint32_t n = 0; n < fromDim[0]; n++) { in convertNchwToNhwc()
111 for (uint32_t c = 0; c < fromDim[1]; c++) { in convertNchwToNhwc()
112 uint32_t fromIndex = n * fromDim[1] * spatialSize + c * spatialSize + hw; in convertNchwToNhwc()
124 const auto& fromDim = nhwcShape.dimensions; in convertNhwcToNchw() local
126 uint32_t spatialSize = fromDim[1] * fromDim[2]; in convertNhwcToNchw()
127 for (uint32_t n = 0; n < fromDim[0]; n++) { in convertNhwcToNchw()
128 for (uint32_t c = 0; c < fromDim[3]; c++) { in convertNhwcToNchw()
130 uint32_t fromIndex = n * spatialSize * fromDim[3] + hw * fromDim[3] + c; in convertNhwcToNchw()