Searched refs:num_rhs_cols (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 183 int64 num_rhs_cols = matrix_b.dim_size(rank - 1); in Compute() local
197 std::swap(num_rhs_rows, num_rhs_cols); in Compute()
206 ctx, AllocateOutput(ctx, rank, batch_size, num_lhs_rows, num_rhs_cols, in Compute()
297 const int64 num_rhs_cols = rhs.dim_size(rhs.dims() - 1); in SparseDenseMatMulWithoutTransposedLHS() local
319 num_rhs_cols, in SparseDenseMatMulWithoutTransposedLHS()
320 num_rhs_rows, num_rhs_cols); in SparseDenseMatMulWithoutTransposedLHS()
325 batch_idx * num_lhs_rows * num_rhs_cols + in SparseDenseMatMulWithoutTransposedLHS()
326 row_begin * num_rhs_cols, in SparseDenseMatMulWithoutTransposedLHS()
327 num_shard_rows, num_rhs_cols); in SparseDenseMatMulWithoutTransposedLHS()
345 const int64 num_rhs_cols = rhs.dim_size(rhs.dims() - 1); in SparseDenseMatMulWithTransposedLHS() local
[all …]
|