Method ParseReplicaGroupsOnly

tensorflow/compiler/xla/service/hlo_parser.cc:2120–2129

replica_groups ::='{' int64list_elements '}' int64list_elements ::= /*empty*/ ::= int64list (',' int64list) int64list ::= '{' int64_elements '}' int64_elements ::= /*empty*/ ::= int64_val (',' int64_val)

//   ::= /*empty*/
//   ::= int64_val (',' int64_val)*
bool HloParserImpl::ParseReplicaGroupsOnly(
    std::vector<ReplicaGroup>* replica_groups) {
  std::vector<std::vector<int64>> result;
  if (!ParseInt64ListList(TokKind::kLbrace, TokKind::kRbrace, TokKind::kComma,
                          &result)) {
    return false;
  }
  *replica_groups = CreateReplicaGroups(result);
  return true;
}

// domain ::= '{' 'kind=' domain_kind ',' 'entry=' entry_sharding ','
//            'exit=' exit_sharding '}'