Method ParseLayout

tensorflow/compiler/xla/service/hlo_parser.cc:3667–3728

layout ::= '{' int64_list (':' tiles element_size_in_bits memory_space)? '}' element_size_in_bits ::= /*empty*/ ::= 'E' '(' int64 ')' memory_space ::= /*empty*/ ::= 'S' '(' int64 ')'

//   ::= /*empty*/
//   ::= 'S' '(' int64 ')'
bool HloParserImpl::ParseLayout(Layout* layout) {
  std::vector<int64> minor_to_major;
  std::vector<Tile> tiles;
  tensorflow::int64 element_size_in_bits = 0;
  tensorflow::int64 memory_space = 0;

  auto parse_and_add_item = [&]() {
    int64 i;
    if (!ParseInt64(&i)) {
      return false;
    }
    minor_to_major.push_back(i);
    return true;
  };

  if (!ParseToken(TokKind::kLbrace,
                  StrCat("expects layout to start with ",
                         TokKindToString(TokKind::kLbrace)))) {
    return false;
  }
  if (lexer_.GetKind() != TokKind::kRbrace) {
    if (lexer_.GetKind() == TokKind::kInt) {
      // Parse minor to major.
      do {
        if (!parse_and_add_item()) {
          return false;
        }
      } while (EatIfPresent(TokKind::kComma));
    }

    if (lexer_.GetKind() == TokKind::kColon) {
      lexer_.Lex();
      if (lexer_.GetKind() == TokKind::kIdent && lexer_.GetStrVal() == "T") {
        lexer_.Lex();
        ParseTiles(&tiles);
      }

      if (lexer_.GetKind() == TokKind::kIdent && lexer_.GetStrVal() == "E") {
        lexer_.Lex();
        ParseLayoutIntAttribute(&element_size_in_bits, "element size in bits");
      }

      if (lexer_.GetKind() == TokKind::kIdent && lexer_.GetStrVal() == "S") {
        lexer_.Lex();
        ParseLayoutIntAttribute(&memory_space, "memory space");
      }
    }
  }
  if (!ParseToken(TokKind::kRbrace,
                  StrCat("expects layout to end with ",
                         TokKindToString(TokKind::kRbrace)))) {
    return false;
  }

  std::vector<Tile> vec_tiles(tiles.size());
  for (int i = 0; i < tiles.size(); i++) {
    vec_tiles[i] = Tile(tiles[i]);
  }