| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 80 const DeviceAssignment* device_assignment) { in set_device_assignment()
85 const DeviceAssignment* ExecutableRunOptions::device_assignment() const { in device_assignment() function in xla::ExecutableRunOptions
|
| D | xla.proto | 305 DeviceAssignmentProto device_assignment = 7; field
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 517 device_assignment=None, argument
664 device_assignment=None, argument
1092 device_assignment=None, argument
1242 device_assignment=None, argument
1319 device_assignment=None, argument
1374 device_assignment=None, argument
1517 device_assignment=None, argument
|
| D | tpu_feed.py | 622 device_assignment=None, argument
746 device_assignment, argument
|
| D | device_assignment.py | 177 def device_assignment(topology, function
|
| D | tpu_context.py | 135 def device_assignment(self): member in TPUContext
338 def device_assignment(self): member in _InternalTPUContext
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compiler.h | 93 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment()
|
| D | hlo_module_config.h | 115 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment()
|
| D | hlo_runner.cc | 274 DeviceAssignment* device_assignment, bool use_threads) { in ExecuteReplicated()
443 int64 device, se::Stream* stream, DeviceAssignment* device_assignment) { in GetServiceRunOptionsForDevice()
|
| D | service.cc | 412 DeviceAssignment device_assignment(options_.number_of_replicas(), in ExecuteParallelAndRegisterResult() local
533 DeviceAssignment device_assignment(options_.number_of_replicas(), in ExecuteAndRegisterResult() local
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt.proto | 49 DeviceAssignment device_assignment = 6; field
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 131 device_assignment=None): argument
199 device_assignment=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.cc | 222 int64 num_replicas, DeviceAssignment* device_assignment, in ExecuteReplicated()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 212 xla::DeviceAssignment device_assignment, xla::LocalClient* client) in LocalExecutable()
|