Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy.py66 elif isinstance(colocate_with, numpy_dataset.SingleDevice):
91 numpy_input, numpy_dataset.SingleDevice(self._input_device), session)
Dcollective_all_reduce_strategy.py123 self._host_input_device = numpy_dataset.SingleDevice(self._worker_device)
183 self._host_input_device = numpy_dataset.SingleDevice(self._worker_device)
242 elif isinstance(colocate_with, numpy_dataset.SingleDevice):
Dparameter_server_strategy.py150 self._input_host_device = numpy_dataset.SingleDevice(worker_device)
209 self._input_host_device = numpy_dataset.SingleDevice(worker_device)
352 if isinstance(colocate_with, numpy_dataset.SingleDevice):
Dnumpy_dataset.py93 class SingleDevice(object): class
Dmirrored_strategy.py475 self._host_input_device = numpy_dataset.SingleDevice("/cpu:0")
504 self._host_input_device = numpy_dataset.SingleDevice(workers[0])
525 elif isinstance(colocate_with, numpy_dataset.SingleDevice):
Dtpu_strategy.py278 numpy_input, numpy_dataset.SingleDevice(self._host_device),
403 elif isinstance(colocate_with, numpy_dataset.SingleDevice):
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition_test.cc240 TEST_F(GraphPartitionTest, SingleDevice) { in TEST_F() argument