Searched refs:spatialSize (Results 1 – 2 of 2) sorted by relevance
127 uint32_t spatialSize = fromDim[2] * fromDim[3]; in convertNchwToNhwc() local129 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertNchwToNhwc()131 uint32_t fromIndex = n * fromDim[1] * spatialSize + c * spatialSize + hw; in convertNchwToNhwc()145 uint32_t spatialSize = fromDim[1] * fromDim[2]; in convertNhwcToNchw() local148 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertNhwcToNchw()149 uint32_t fromIndex = n * spatialSize * fromDim[3] + hw * fromDim[3] + c; in convertNhwcToNchw()
408 uint32_t spatialSize = fromDim[2] * fromDim[3]; in convertToNhwcImpl() local410 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertToNhwcImpl()412 uint32_t fromIndex = n * fromDim[1] * spatialSize + c * spatialSize + hw; in convertToNhwcImpl()422 uint32_t spatialSize = fromDim[1] * fromDim[2]; in convertFromNhwcImpl() local425 for (uint32_t hw = 0; hw < spatialSize; hw++) { in convertFromNhwcImpl()426 uint32_t fromIndex = n * spatialSize * fromDim[3] + hw * fromDim[3] + c; in convertFromNhwcImpl()