Searched refs:c_matrix_rhs (Results 1 – 1 of 1) sorted by relevance
535 const int64 c_matrix_rhs = c_shape.dim_size(row_dim + 1); in Compute() local536 const int64 c_slice_size = c_matrix_lhs * c_matrix_rhs; in Compute()601 c_mat_col_major_shape = TensorShape({c_matrix_rhs, c_matrix_lhs}); in Compute()604 TensorShape({batch_size, c_matrix_rhs, c_matrix_lhs}); in Compute()671 {c_matrix_lhs, c_matrix_rhs}); in Compute()