10 #ifndef EIGEN_CXX11_TENSOR_TENSOR_CONTRACTION_BLOCKING_H
11 #define EIGEN_CXX11_TENSOR_TENSOR_CONTRACTION_BLOCKING_H
24 template<
typename ResScalar,
typename LhsScalar,
typename RhsScalar,
typename StorageIndex,
int ShardingType = ShardByCol>
42 #if !defined(EIGEN_HIPCC)
49 computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(
kc_,
mc_,
nc_, num_threads);
52 computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(
kc_,
nc_,
mc_, num_threads);
56 kc_ = (rhs_packet_size <= 8 ||
kc_ <= rhs_packet_size) ?
57 kc_ : (
kc_ / rhs_packet_size) * rhs_packet_size;
73 #endif // EIGEN_CXX11_TENSOR_TENSOR_CONTRACTION_BLOCKING_H