| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | computation_placer.h | 38 class DeviceAssignment : public Array2D<int> {
40 DeviceAssignment() {} in DeviceAssignment() function
41 DeviceAssignment(int replica_count, int computation_count) in DeviceAssignment() function
56 static StatusOr<std::unique_ptr<DeviceAssignment>> Deserialize(
77 virtual StatusOr<DeviceAssignment> AssignDevices(int replica_count,
|
| D | computation_placer.cc | 41 Status DeviceAssignment::Serialize(DeviceAssignmentProto* proto) const { in Serialize()
54 /* static */ StatusOr<std::unique_ptr<DeviceAssignment>>
55 DeviceAssignment::Deserialize(const DeviceAssignmentProto& proto) { in Deserialize()
63 auto assignment = absl::make_unique<DeviceAssignment>( in Deserialize()
78 string DeviceAssignment::ToString() const { in ToString()
100 StatusOr<DeviceAssignment> ComputationPlacer::AssignDevices( in AssignDevices()
102 DeviceAssignment assignment(replica_count, computation_count); in AssignDevices()
|
| D | hlo_module_config.h | 111 const DeviceAssignment& static_device_assignment() const { in static_device_assignment()
115 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment()
137 absl::optional<DeviceAssignment> static_device_assignment_;
|
| D | compiler.h | 89 const DeviceAssignment& static_device_assignment() const { in static_device_assignment()
93 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment()
103 absl::optional<DeviceAssignment> static_device_assignment_;
|
| D | hlo_runner.h | 180 DeviceAssignment* device_assignment, bool use_threads = false);
196 int64 device, se::Stream* stream, DeviceAssignment* device_assignment);
|
| D | hlo_runner.cc | 274 DeviceAssignment* device_assignment, bool use_threads) { in ExecuteReplicated()
420 DeviceAssignment device_assignment, in ExecuteReplicated()
443 int64 device, se::Stream* stream, DeviceAssignment* device_assignment) { in GetServiceRunOptionsForDevice()
|
| D | service.cc | 328 DeviceAssignment::Deserialize(execution_options->device_assignment())); in CreateModuleConfig()
412 DeviceAssignment device_assignment(options_.number_of_replicas(), in ExecuteParallelAndRegisterResult()
533 DeviceAssignment device_assignment(options_.number_of_replicas(), in ExecuteAndRegisterResult()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.h | 35 class DeviceAssignment; variable
77 const DeviceAssignment* device_assignment);
78 const DeviceAssignment* device_assignment() const;
86 const DeviceAssignment* device_assignment_ = nullptr;
|
| D | executable_run_options.cc | 80 const DeviceAssignment* device_assignment) { in set_device_assignment()
85 const DeviceAssignment* ExecutableRunOptions::device_assignment() const { in device_assignment()
|
| D | xla_data.proto | 339 // DeviceAssignmentProto is a serialized form of DeviceAssignment class, which
341 // See xla::DeviceAssignment class comment for more details.
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.tpu.experimental.-device-assignment.pbtxt | 1 path: "tensorflow.tpu.experimental.DeviceAssignment"
3 is_instance: "<class \'tensorflow.python.tpu.device_assignment.DeviceAssignment\'>"
|
| D | tensorflow.tpu.experimental.pbtxt | 4 name: "DeviceAssignment"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.tpu.experimental.-device-assignment.pbtxt | 1 path: "tensorflow.tpu.experimental.DeviceAssignment"
3 is_instance: "<class \'tensorflow.python.tpu.device_assignment.DeviceAssignment\'>"
|
| D | tensorflow.tpu.experimental.pbtxt | 4 name: "DeviceAssignment"
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt.proto | 10 message DeviceAssignment { message
48 // DeviceAssignment proto.
49 DeviceAssignment device_assignment = 6;
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | device_assignment.py | 58 class DeviceAssignment(object): class
323 return DeviceAssignment(topology, core_assignment=assignment)
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.h | 127 xla::DeviceAssignment device_assignment,
148 const xla::DeviceAssignment device_assignment_;
|
| D | local_computation_builder.cc | 212 xla::DeviceAssignment device_assignment, xla::LocalClient* client) in LocalExecutable()
356 DeviceAssignment device_assignment, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.h | 188 int64 num_replicas, DeviceAssignment* device_assignment,
|
| D | hlo_test_base.cc | 222 int64 num_replicas, DeviceAssignment* device_assignment, in ExecuteReplicated()
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | combinations.py | 336 device_assignment = device_assignment_lib.DeviceAssignment(
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 217 device_assignment_lib.DeviceAssignment)
|