Searched refs:num_lhs_rows (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 180 int64 num_lhs_rows = dense_shape(rank - 2); in Compute() local
186 std::swap(num_lhs_rows, num_lhs_cols); in Compute()
206 ctx, AllocateOutput(ctx, rank, batch_size, num_lhs_rows, num_rhs_cols, in Compute()
212 ctx, batch_size, num_lhs_rows, *sparse_matrix_a, *rhs, matmul_result); in Compute()
214 SparseDenseMatMulWithTransposedLHS(ctx, batch_size, num_lhs_rows, in Compute()
288 OpKernelContext* ctx, const int64 batch_size, const int64 num_lhs_rows, in SparseDenseMatMulWithoutTransposedLHS() argument
295 num_lhs_rows / std::max(kMaxShards, kNumShardsPerThread * num_threads); in SparseDenseMatMulWithoutTransposedLHS()
299 batch_size * num_lhs_rows /* total */, in SparseDenseMatMulWithoutTransposedLHS()
306 num_lhs_rows, batch_and_row_begin, batch_and_row_end, in SparseDenseMatMulWithoutTransposedLHS()
325 batch_idx * num_lhs_rows * num_rhs_cols + in SparseDenseMatMulWithoutTransposedLHS()
[all …]
|