Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc544 const int64_t c_matrix_rhs = c_shape.dim_size(row_dim + 1); in Compute() local
545 const int64_t c_slice_size = c_matrix_lhs * c_matrix_rhs; in Compute()
610 c_mat_col_major_shape = TensorShape({c_matrix_rhs, c_matrix_lhs}); in Compute()
613 TensorShape({batch_size, c_matrix_rhs, c_matrix_lhs}); in Compute()
680 {c_matrix_lhs, c_matrix_rhs}); in Compute()