Searched refs:host_disconnect_def (Results 1 – 1 of 1) sorted by relevance
215 NodeDef host_disconnect_def; in AddHostDisconnectNode() local216 host_disconnect_def.set_name(graph->NewName("disconnect_tpu_host")); in AddHostDisconnectNode()217 host_disconnect_def.set_op(kHostDisconnectOp); in AddHostDisconnectNode()218 host_disconnect_def.set_device(host_device_name); in AddHostDisconnectNode()220 &host_disconnect_def); in AddHostDisconnectNode()223 Node* host_disconnect_node = graph->AddNode(host_disconnect_def, &status); in AddHostDisconnectNode()