10 #ifndef EIGEN_CXX11_TENSOR_TENSOR_CONTRACTION_BLOCKING_H
11 #define EIGEN_CXX11_TENSOR_TENSOR_CONTRACTION_BLOCKING_H
24 template <
typename LhsMapper,
typename RhsMapper,
typename Index,
int ShardingType=ShardByCol>
25 class TensorContractionBlocking {
28 typedef typename LhsMapper::Scalar LhsScalar;
29 typedef typename RhsMapper::Scalar RhsScalar;
32 kc_(k), mc_(m), nc_(n)
34 if (ShardingType == ShardByCol) {
35 computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(kc_, mc_, nc_, num_threads);
38 computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(kc_, nc_, mc_, num_threads);
42 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
Index kc()
const {
return kc_; }
43 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
Index mc()
const {
return mc_; }
44 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
Index nc()
const {
return nc_; }
Namespace containing all symbols from the Eigen library.
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index