Searched refs:ShardByCol (Results 1 – 4 of 4) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContractionBlocking.h | 19 ShardByCol = 1 enumerator
24 template <typename LhsMapper, typename RhsMapper, typename Index, int ShardingType=ShardByCol>
34 if (ShardingType == ShardByCol) { in kc_()
|
| D | TensorContractionThreadPool.h | 185 internal::ShardByCol>
229 internal::ShardByCol>
797 …internal::TensorContractionBlocking<LhsMapper, RhsMapper, Index, internal::ShardByCol> blocking(k,…
|
| D | TensorContraction.h | 481 …internal::TensorContractionBlocking<LhsMapper, RhsMapper, Index, internal::ShardByCol> blocking(k,…
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_contraction_kernel.h | 195 if (sharding_type == ShardByCol) {
|