Lines Matching refs:fromDim

490 inline bool convertToNhwcImpl(T* to, const T* from, const std::vector<uint32_t>& fromDim) {  in convertToNhwcImpl()  argument
491 uint32_t spatialSize = fromDim[2] * fromDim[3]; in convertToNhwcImpl()
492 for (uint32_t n = 0; n < fromDim[0]; n++) { in convertToNhwcImpl()
494 for (uint32_t c = 0; c < fromDim[1]; c++) { in convertToNhwcImpl()
495 uint32_t fromIndex = n * fromDim[1] * spatialSize + c * spatialSize + hw; in convertToNhwcImpl()
504 inline bool convertFromNhwcImpl(T* to, const T* from, const std::vector<uint32_t>& fromDim) { in convertFromNhwcImpl() argument
505 uint32_t spatialSize = fromDim[1] * fromDim[2]; in convertFromNhwcImpl()
506 for (uint32_t n = 0; n < fromDim[0]; n++) { in convertFromNhwcImpl()
507 for (uint32_t c = 0; c < fromDim[3]; c++) { in convertFromNhwcImpl()
509 uint32_t fromIndex = n * spatialSize * fromDim[3] + hw * fromDim[3] + c; in convertFromNhwcImpl()
528 auto& fromDim = from.dimensions; in convertToNhwc() local
529 inShape.dimensions = {fromDim[0], fromDim[2], fromDim[3], fromDim[1]}; in convertToNhwc()
539 reinterpret_cast<const float*>(from.buffer), fromDim); in convertToNhwc()
543 fromDim); in convertToNhwc()
547 fromDim); in convertToNhwc()
550 reinterpret_cast<const int8_t*>(from.buffer), fromDim); in convertToNhwc()
570 auto& fromDim = from.dimensions; in convertFromNhwc() local
571 outShape.dimensions = {fromDim[0], fromDim[3], fromDim[1], fromDim[2]}; in convertFromNhwc()
579 reinterpret_cast<const float*>(from.buffer), fromDim); in convertFromNhwc()
583 fromDim); in convertFromNhwc()
587 fromDim); in convertFromNhwc()
591 fromDim); in convertFromNhwc()