get_host_for_device (reference) in projects: external

Searched refs:get_host_for_device (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
D	device_util.py	`122 def get_host_for_device(device): function`
D	one_device_strategy.py	`261 self._input_device = device_util.get_host_for_device(self._device)`
D	collective_all_reduce_strategy.py	`524 host_device = device_util.get_host_for_device(self._worker_device) 530 [device_util.get_host_for_device(worker) for worker in`
D	tpu_strategy.py	`836 self._host_device = device_util.get_host_for_device(self._tpu_devices[0][0]) 845 host_device = device_util.get_host_for_device(tpu_device)`
D	input_lib.py	`1764 host_device = device_util.get_host_for_device(self._worker)`
D	input_lib_test.py	`453 host_device = device_util.get_host_for_device(tpu_device)`
/external/tensorflow/tensorflow/python/tpu/
D	tpu_embedding_v2.py	`1309 with ops.device(device_util.get_host_for_device(tpu_device)): 1325 with ops.device(device_util.get_host_for_device(device)): 1656 host = device_util.get_host_for_device(tpu_device)`