MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / packLhs

Function packLhs

tensorflow/core/kernels/eigen_contraction_kernel.h:612–630  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

610 } \
611 \
612 EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE void packLhs( \
613 LhsBlock* lhsBlock, const typename LhsMapper::SubMapper& data_mapper, \
614 const StorageIndex depth, const StorageIndex rows) { \
615 if (UseCustomContractionKernels()) { \
616 const bool is_direct_access = \
617 DirectLhsAccess::value && \
618 DirectLhsAccess::block(data_mapper, rows, depth, \
619 bn > 0 ? divup(n, bn) : 0, lhsBlock); \
620 \
621 if (!is_direct_access) { \
622 lhsBlock->is_direct_access = false; \
623 LhsPacker()(lhsBlock->packed_data, data_mapper, rows, depth); \
624 } \
625 } else { \
626 lhsBlock->is_direct_access = false; \
627 EigenLhsPacker()(lhsBlock->packed_data, data_mapper, depth, rows, \
628 /*stride*/ 0, /*offset*/ 0); \
629 } \
630 } \
631 \
632 EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE void packRhs( \
633 RhsBlock* rhsBlock, const typename RhsMapper::SubMapper& data_mapper, \

Callers

nothing calls this directly

Calls 3

blockFunction · 0.85
divupFunction · 0.85

Tested by

no test coverage detected