Home
last modified time | relevance | path

Searched refs:kMaxShards (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc56 static constexpr int32 kMaxShards = 20; variable
295 num_lhs_rows / std::max(kMaxShards, kNumShardsPerThread * num_threads); in SparseDenseMatMulWithoutTransposedLHS()
368 num_lhs_cols / std::max(kMaxShards, kNumShardsPerThread * num_threads); in SparseDenseMatMulWithTransposedLHS()