Searched refs:spatialSize (Results 1 – 2 of 2) sorted by relevance
108 uint32_t spatialSize = fromDim[2] * fromDim[3]; in convertNchwToNhwc() local110 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertNchwToNhwc()112 uint32_t fromIndex = n * fromDim[1] * spatialSize + c * spatialSize + hw; in convertNchwToNhwc()126 uint32_t spatialSize = fromDim[1] * fromDim[2]; in convertNhwcToNchw() local129 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertNhwcToNchw()130 uint32_t fromIndex = n * spatialSize * fromDim[3] + hw * fromDim[3] + c; in convertNhwcToNchw()
424 uint32_t spatialSize = fromDim[2] * fromDim[3]; in convertToNhwcImpl() local426 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertToNhwcImpl()428 uint32_t fromIndex = n * fromDim[1] * spatialSize + c * spatialSize + hw; in convertToNhwcImpl()438 uint32_t spatialSize = fromDim[1] * fromDim[2]; in convertFromNhwcImpl() local441 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertFromNhwcImpl()442 uint32_t fromIndex = n * spatialSize * fromDim[3] + hw * fromDim[3] + c; in convertFromNhwcImpl()