Searched refs:spatialSize (Results 1 – 2 of 2) sorted by relevance
126 uint32_t spatialSize = fromDim[2] * fromDim[3]; in convertNchwToNhwc() local128 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertNchwToNhwc()130 uint32_t fromIndex = n * fromDim[1] * spatialSize + c * spatialSize + hw; in convertNchwToNhwc()144 uint32_t spatialSize = fromDim[1] * fromDim[2]; in convertNhwcToNchw() local147 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertNhwcToNchw()148 uint32_t fromIndex = n * spatialSize * fromDim[3] + hw * fromDim[3] + c; in convertNhwcToNchw()
489 uint32_t spatialSize = fromDim[2] * fromDim[3]; in convertToNhwcImpl() local491 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertToNhwcImpl()493 uint32_t fromIndex = n * fromDim[1] * spatialSize + c * spatialSize + hw; in convertToNhwcImpl()503 uint32_t spatialSize = fromDim[1] * fromDim[2]; in convertFromNhwcImpl() local506 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertFromNhwcImpl()507 uint32_t fromIndex = n * spatialSize * fromDim[3] + hw * fromDim[3] + c; in convertFromNhwcImpl()