/external/tensorflow/tensorflow/compiler/xla/service/ |
D | collective_ops_utils_test.cc | 53 DeviceAssignment device_assignment(/*replica_count=*/3, in TEST() local 69 DeviceAssignment device_assignment(/*replica_count=*/4, in TEST() local 92 DeviceAssignment device_assignment(/*replica_count=*/2, in TEST() local
|
D | hlo_runner.cc | 185 DeviceAssignment* device_assignment) { in ExecuteReplicated() 200 DeviceAssignment* device_assignment) { in ExecuteReplicatedImpl() 321 DeviceAssignment* device_assignment, ExecutionProfile* profile) { in ExecuteReplicated() 443 int64 device, se::Stream* stream, DeviceAssignment* device_assignment, in GetServiceRunOptionsForDevice()
|
D | collective_ops_utils.cc | 79 GlobalDeviceId device_id, const DeviceAssignment& device_assignment, in GetParticipatingDevices()
|
D | compiler.h | 103 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment()
|
D | hlo_module_config.h | 185 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment()
|
D | service.cc | 362 DeviceAssignment device_assignment(options_.number_of_replicas(), in ExecuteParallelAndRegisterResult() local 485 DeviceAssignment device_assignment(options_.number_of_replicas(), in ExecuteAndRegisterResult() local
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | executable_run_options.cc | 96 const DeviceAssignment* device_assignment) { in set_device_assignment() 101 const DeviceAssignment* ExecutableRunOptions::device_assignment() const { in device_assignment() function in xla::ExecutableRunOptions
|
/external/tensorflow/tensorflow/core/tpu/kernels/ |
D | tpu_compile_op_support.cc | 59 absl::optional<const DeviceAssignment> device_assignment, int replica_count, in CreateModuleConfig() 138 absl::optional<const DeviceAssignment> device_assignment, int replica_count, in CreateModuleConfig() 310 const absl::optional<xla::DeviceAssignment>& device_assignment, in CreateHloModules() 422 const DeviceAssignment& device_assignment = in CompileOpMetadataFromContext() local
|
D | tpu_executable_info.proto | 63 xla.DeviceAssignmentProto device_assignment = 6; field
|
/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
D | xrt_compile_ops.cc | 56 int num_cores_per_replica, xla::DeviceAssignment* device_assignment) { in GenerateXlaDeviceAssignment() 170 xla::DeviceAssignment device_assignment(num_replicas, in Compile() local
|
/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
D | gpu_multistream_test.cc | 53 DeviceAssignment device_assignment(1, 1); in TEST() local
|
D | utils.cc | 118 std::shared_ptr<DeviceAssignment>* device_assignment) { in ParseDeviceAssignmentCompileOptions()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | executable_build_options.cc | 93 const DeviceAssignment& device_assignment) { in set_device_assignment()
|
D | executable_build_options.h | 108 const DeviceAssignment& device_assignment() const { in device_assignment() function
|
/external/tensorflow/tensorflow/python/tpu/ |
D | tpu_feed.py | 647 device_assignment=None, argument 771 device_assignment, argument
|
D | device_assignment.py | 337 def device_assignment( function
|
/external/tensorflow/tensorflow/core/protobuf/tpu/ |
D | compile_metadata.proto | 86 xla.DeviceAssignmentProto device_assignment = 8; field
|
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
D | libtpu_client.c | 117 DeviceAssignment device_assignment = {NULL, 0}; in main() local
|
D | tpu_client.cc | 511 DeviceAssignment device_assignment, std::shared_ptr<PyTpuClient> client, in PyTpuExecutable() 581 xla::DeviceAssignmentProto device_assignment; in ExecuteHelper() local 810 absl::optional<xla::DeviceAssignment> device_assignment; in Compile() local
|
D | tpu_client.h | 322 const DeviceAssignment& device_assignment() const { in device_assignment() function
|
/external/tensorflow/tensorflow/python/distribute/ |
D | tpu_strategy.py | 627 device_assignment=None): argument 687 device_assignment=None): argument 790 device_assignment=None): argument
|
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ |
D | tpu_service.proto | 94 optional xla.DeviceAssignmentProto device_assignment = 4; field
|
/external/tensorflow/tensorflow/compiler/xrt/ |
D | xrt.proto | 50 DeviceAssignment device_assignment = 6; field
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | outfeed_receiver.cc | 402 DeviceAssignment device_assignment(1, 1); in SendShutdownOutfeedHeader() local
|
/external/tensorflow/tensorflow/core/tpu/ |
D | tpu_execute.cc | 418 TpuNodeContext* node_context, xla::DeviceAssignment* device_assignment, in TPUExecute()
|