Searched refs:rhs_delete_dims (Results 1 – 1 of 1) sorted by relevance
411 absl::InlinedVector<int64, 8> rhs_delete_dims; in Einsum() local426 rhs_delete_dims.push_back(rhs_dim); in Einsum()437 rhs_delete_dims.push_back(rhs_dim); in Einsum()479 rhs_delete_dims.push_back(d); in Einsum()495 absl::c_sort(rhs_delete_dims); in Einsum()496 DeleteDimsFromContainer(rhs_delete_dims, &y_shape, in Einsum()505 if (!rhs_delete_dims.empty()) { in Einsum()508 rhs_delete_dims); in Einsum()