Lines Matching refs:fromDim

488 inline bool convertToNhwcImpl(T* to, const T* from, const std::vector<uint32_t>& fromDim) {  in convertToNhwcImpl()  argument
489 uint32_t spatialSize = fromDim[2] * fromDim[3]; in convertToNhwcImpl()
490 for (uint32_t n = 0; n < fromDim[0]; n++) { in convertToNhwcImpl()
492 for (uint32_t c = 0; c < fromDim[1]; c++) { in convertToNhwcImpl()
493 uint32_t fromIndex = n * fromDim[1] * spatialSize + c * spatialSize + hw; in convertToNhwcImpl()
502 inline bool convertFromNhwcImpl(T* to, const T* from, const std::vector<uint32_t>& fromDim) { in convertFromNhwcImpl() argument
503 uint32_t spatialSize = fromDim[1] * fromDim[2]; in convertFromNhwcImpl()
504 for (uint32_t n = 0; n < fromDim[0]; n++) { in convertFromNhwcImpl()
505 for (uint32_t c = 0; c < fromDim[3]; c++) { in convertFromNhwcImpl()
507 uint32_t fromIndex = n * spatialSize * fromDim[3] + hw * fromDim[3] + c; in convertFromNhwcImpl()
526 auto& fromDim = from.dimensions; in convertToNhwc() local
527 inShape.dimensions = {fromDim[0], fromDim[2], fromDim[3], fromDim[1]}; in convertToNhwc()
537 reinterpret_cast<const float*>(from.buffer), fromDim); in convertToNhwc()
541 fromDim); in convertToNhwc()
545 fromDim); in convertToNhwc()
548 reinterpret_cast<const int8_t*>(from.buffer), fromDim); in convertToNhwc()
568 auto& fromDim = from.dimensions; in convertFromNhwc() local
569 outShape.dimensions = {fromDim[0], fromDim[3], fromDim[1], fromDim[2]}; in convertFromNhwc()
577 reinterpret_cast<const float*>(from.buffer), fromDim); in convertFromNhwc()
581 fromDim); in convertFromNhwc()
585 fromDim); in convertFromNhwc()
589 fromDim); in convertFromNhwc()