Searched refs:wait_def (Results 1 – 1 of 1) sorted by relevance
102 NodeDef wait_def; in AddWaitNode() local103 wait_def.set_name(graph->NewName("wait_for_distributed_tpu_system")); in AddWaitNode()104 wait_def.set_op(kWaitOp); in AddWaitNode()105 wait_def.set_device(configuration_device_name); in AddWaitNode()107 &wait_def); in AddWaitNode()108 AddNodeAttr("startup_timeout_sec", kDefaultStartupTimeout, &wait_def); in AddWaitNode()111 &wait_def); in AddWaitNode()115 *wait_node = graph->AddNode(wait_def, &status); in AddWaitNode()