Searched refs:xla_token_input_nodes (Results 1 – 1 of 1) sorted by relevance
346 std::vector<string> xla_token_input_nodes; in BuildXlaHostComputeNodeDef() local347 xla_token_input_nodes.emplace_back(kXlaTokenArgNodeName); in BuildXlaHostComputeNodeDef()351 xla_token_input_nodes.emplace_back(host_compute_node_name(dep)); in BuildXlaHostComputeNodeDef()354 host_compute_builder.Attr(kXlaTokenInputNodesAttrName, xla_token_input_nodes); in BuildXlaHostComputeNodeDef()