func.return
}

// CHECK-LABEL: @CommunicateOpTokenAttrs
func.func @CommunicateOpTokenAttrs() -> () {
  // CHECK: _xla_original_oc_node_name = [[NODE_NAME1:.*]], _xla_token_input_nodes = ["_xla_token_arg_node"]
  %0 = "tf.XlaRecvFromHost"() {key = "recv_key", shape = #tf_type.shape<>} : () -> tensor<i32>

  // CHECK: _xla_original_oc_node_name = [[NODE_NAME2:.*]], _xla_token_input_nodes = {{\[}}[[NODE_NAME1]]{{\]}}

  func.func @main() {
    tf_executor.graph {
      %control = tf_executor.island wraps "tf.XlaHostCompute"() {_xla_original_oc_node_name = "Add", _xla_token_input_nodes = ["_xla_token_arg_node"], ancestors = [], cost_estimate_ns = 1024 : i64, key = "host_compute_channel_1_retvals", send_key = "", recv_key = "", shapes = [], tpu_core = 0 : i64} : () -> ()
      tf_executor.fetch
    }
    func.return
  }
}

            {"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"}});

  {

                          "_xla_token_input_nodes", &token_input_nodes));

  std::vector<string> expected_token_input_nodes_0({"_xla_token_arg_node"});
  EXPECT_EQ(token_input_nodes, expected_token_input_nodes_0);
  token_input_nodes.clear();
  std::vector<string> expected_token_input_nodes_1(
      {"_xla_token_arg_node", "outside_compilation_0_host_compute"});
  TF_CHECK_OK(GetNodeAttr(AttrSlice(host_compute_1->attrs()),

    // CHECK-SAME:_xla_original_oc_node_name = "hcb0", _xla_token_input_nodes = ["_xla_token_arg_node"] 
    "tf.XlaHostCompute"(%cst) <{ancestors = [], cost_estimate_ns = 1000000 : i64, key = "_host_callback", recv_key = "", send_key = "", shapes = [], tpu_core = 0 : i64}> {_xla_original_oc_node_name = "hcb0", _xla_token_input_nodes = ["_xla_token_arg_node"]} : (tensor<i32>) -> ()
    tf_device.return

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