Searched refs:col_dim (Results 1 – 1 of 1) sorted by relevance
192 int64 col_dim = dim_nums.lhs_batch_dimensions_size() + 1; in RunGemm() local199 CHECK_NE(col_dim, batch_dim); in RunGemm()206 CHECK_LT(shape->layout().minor_to_major(col_dim), 2); in RunGemm()210 int64 output_num_cols = output_shape.dimensions(col_dim); in RunGemm()246 rhs_buffer, rhs_shape, dim_nums.rhs_contracting_dimensions(0) == col_dim); in RunGemm()