int used_rank = tensor_shape.getRank();

  if (is_lhs) {
    used_rank -= dnums.lhs_contracting_dimensions_size();
    for (int64_t v : dnums.lhs_contracting_dimensions()) {
      opponent_shape.push_back(tensor_shape.getDimSize(v));
    }
  } else {
    used_rank -= dnums.rhs_contracting_dimensions_size();
    for (int64_t v : dnums.rhs_contracting_dimensions()) {