{"shapes", absl::Span<const DataType>({})},
            {"_outside_compilation_subgraph", "O1"},
            {"_xla_token_input_nodes",
             absl::Span<const string>({"_xla_token_arg_node"})},
            {"_xla_original_oc_node_name",
             "outside_compilation_O1_host_compute"}},
           {"c"}},
      },
      {{"f_0_retval_retval", "F:o:0"}});

  {

    func.return %2336#0, %2336#1, %2336#2, %2336#3: tensor<1x1120x8xi32>, tensor<1x1120x8xi32>,...