Searched refs:num_lhs_cols (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 181 int64 num_lhs_cols = dense_shape(rank - 1); in Compute() local
186 std::swap(num_lhs_rows, num_lhs_cols); in Compute()
215 num_lhs_cols, *sparse_matrix_a, *rhs, in Compute()
338 const int64 num_lhs_cols, in SparseDenseMatMulWithTransposedLHS() argument
368 num_lhs_cols / std::max(kMaxShards, kNumShardsPerThread * num_threads); in SparseDenseMatMulWithTransposedLHS()
370 batch_size * num_lhs_cols /* total */, in SparseDenseMatMulWithTransposedLHS()
377 num_lhs_cols, batch_and_row_begin, batch_and_row_end, in SparseDenseMatMulWithTransposedLHS()
|