Home
last modified time | relevance | path

Searched refs:configuration_device_name (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc55 Status AddConfigurationNode(const string& configuration_device_name, in AddConfigurationNode() argument
62 config_def.set_device(configuration_device_name); in AddConfigurationNode()
73 (*configuration_node)->set_assigned_device_name(configuration_device_name); in AddConfigurationNode()
99 Status AddWaitNode(const string& configuration_device_name, in AddWaitNode() argument
105 wait_def.set_device(configuration_device_name); in AddWaitNode()
119 (*wait_node)->set_assigned_device_name(configuration_device_name); in AddWaitNode()
261 const string& configuration_device_name, in Run()
294 configuration_device_name, host_devices.size(), graph, in Run()
319 TF_RETURN_IF_ERROR(AddWaitNode(configuration_device_name, in Run()
Ddistributed_tpu_rewrite_helpers.h89 const string& configuration_device_name,
Ddistributed_tpu_rewrite_helpers.cc204 const string& configuration_device_name, in ForConfigurationNodeMatchingType() argument