"_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