Searched refs:SingleDevice (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_optimizer.cc | 96 InlinedFunctionBodyPlacer::SingleDevice(); in Optimize()
101 InlinedFunctionBodyPlacer::SingleDevice(); in Optimize()
|
| D | lower_function_call_op.cc | 74 InlinedFunctionBodyPlacer::SingleDevice(); in RewriteFunctionCallNode()
|
| D | inline_function_utils.h | 76 static Config SingleDevice() { return {"single_device", SingleDevicePlacer}; } in SingleDevice() function
|
| D | function_test.cc | 1266 InlinedFunctionBodyPlacer::SingleDevice(); in TEST_F()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | one_device_strategy.py | 279 elif isinstance(colocate_with, numpy_dataset.SingleDevice):
306 numpy_input, numpy_dataset.SingleDevice(self._input_device), session)
|
| D | numpy_dataset.py | 94 class SingleDevice(object): class
|
| D | parameter_server_strategy.py | 225 self._input_host_device = numpy_dataset.SingleDevice(self._worker_device)
289 self._input_host_device = numpy_dataset.SingleDevice(self._worker_device)
477 if isinstance(colocate_with, numpy_dataset.SingleDevice):
|
| D | mirrored_strategy.py | 390 self._host_input_device = numpy_dataset.SingleDevice(
419 self._host_input_device = numpy_dataset.SingleDevice(workers[0])
497 elif isinstance(colocate_with, numpy_dataset.SingleDevice):
|
| D | collective_all_reduce_strategy.py | 370 self._host_input_device = numpy_dataset.SingleDevice(self._worker_device)
431 self._host_input_device = numpy_dataset.SingleDevice(self._worker_device)
|
| D | tpu_strategy.py | 908 numpy_input, numpy_dataset.SingleDevice(self._host_device),
1128 elif isinstance(colocate_with, numpy_dataset.SingleDevice):
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime.cc | 152 InlinedFunctionBodyPlacer::SingleDevice(); in ConstructFunctionGraph()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition_test.cc | 240 TEST_F(GraphPartitionTest, SingleDevice) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 1301 InlinedFunctionBodyPlacer::SingleDevice(); in InlineFunctionCalls()
|