Searched refs:host_device_name (Results 1 – 5 of 5) sorted by relevance
77 Status AddHostConfigNode(const string& host_device_name, in AddHostConfigNode() argument84 host_config_def.set_device(host_device_name); in AddHostConfigNode()94 (*host_configuration_node)->set_assigned_device_name(host_device_name); in AddHostConfigNode()127 Status AddGlobalTPUArrayNode(const string& host_device_name, Node* wait_node, in AddGlobalTPUArrayNode() argument132 global_tpu_array_def.set_device(host_device_name); in AddGlobalTPUArrayNode()140 (*global_tpu_array_node)->set_assigned_device_name(host_device_name); in AddGlobalTPUArrayNode()211 Status AddHostDisconnectNode(const string& host_device_name, in AddHostDisconnectNode() argument218 host_disconnect_def.set_device(host_device_name); in AddHostDisconnectNode()227 host_disconnect_node->set_assigned_device_name(host_device_name); in AddHostDisconnectNode()
3853 string host_device_name; in LowerOutsideCompilationFunctionalNodes() local3855 tpu_device_name, &host_device_name)); in LowerOutsideCompilationFunctionalNodes()3856 n->set_assigned_device_name(host_device_name); in LowerOutsideCompilationFunctionalNodes()
1052 char * host_device_name = NULL; in main() local1153 host_device_name = optarg; in main()1255 if (host_device_name) { in main()1256 host_dev = open (host_device_name, open_flag); in main()1260 host_device_name); in main()
212 std::string* host_device_name);
582 const string& device_name, string* host_device_name) { in DeviceNameToCpuDeviceName() argument591 *host_device_name = DeviceNameUtils::ParsedNameToString(device); in DeviceNameToCpuDeviceName()