Lines Matching refs:fromDim
423 inline bool convertToNhwcImpl(T* to, const T* from, const std::vector<uint32_t>& fromDim) { in convertToNhwcImpl() argument
424 uint32_t spatialSize = fromDim[2] * fromDim[3]; in convertToNhwcImpl()
425 for (uint32_t n = 0; n < fromDim[0]; n++) { in convertToNhwcImpl()
427 for (uint32_t c = 0; c < fromDim[1]; c++) { in convertToNhwcImpl()
428 uint32_t fromIndex = n * fromDim[1] * spatialSize + c * spatialSize + hw; in convertToNhwcImpl()
437 inline bool convertFromNhwcImpl(T* to, const T* from, const std::vector<uint32_t>& fromDim) { in convertFromNhwcImpl() argument
438 uint32_t spatialSize = fromDim[1] * fromDim[2]; in convertFromNhwcImpl()
439 for (uint32_t n = 0; n < fromDim[0]; n++) { in convertFromNhwcImpl()
440 for (uint32_t c = 0; c < fromDim[3]; c++) { in convertFromNhwcImpl()
442 uint32_t fromIndex = n * spatialSize * fromDim[3] + hw * fromDim[3] + c; in convertFromNhwcImpl()
461 auto& fromDim = from.dimensions; in convertToNhwc() local
462 inShape.dimensions = {fromDim[0], fromDim[2], fromDim[3], fromDim[1]}; in convertToNhwc()
472 reinterpret_cast<const float*>(from.buffer), fromDim); in convertToNhwc()
476 fromDim); in convertToNhwc()
480 fromDim); in convertToNhwc()
500 auto& fromDim = from.dimensions; in convertFromNhwc() local
501 outShape.dimensions = {fromDim[0], fromDim[3], fromDim[1], fromDim[2]}; in convertFromNhwc()
509 reinterpret_cast<const float*>(from.buffer), fromDim); in convertFromNhwc()
513 fromDim); in convertFromNhwc()
517 fromDim); in convertFromNhwc()