Home
last modified time | relevance | path

Searched refs:set_static_device_assignment (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_util.cc130 config->set_static_device_assignment(*device_assignment); in CreateModuleConfig()
Dhlo_module_config.h233 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment() function
Dhlo_module.cc517 module->config_.set_static_device_assignment(*device_assignment); in CreateFromProto()
557 module_config.set_static_device_assignment(*device_assignment); in CreateModuleConfigFromShape()
601 config.set_static_device_assignment(*device_assignment); in CreateModuleConfigFromProto()
Dcompiler.h124 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment() function
Dhlo_reachability_test.cc204 module->config().set_static_device_assignment(DeviceAssignment(1, 2)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dall_reduce_blueconnect_test.cc41 module.config().set_static_device_assignment(device_assignment); in SetModuleConfig()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc120 config->set_static_device_assignment(device_assignment.value()); in CreateModuleConfig()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dc_api_conversions.cc521 config.set_static_device_assignment( in FromC()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcollective_ops_test.cc340 config.set_static_device_assignment(device_assn); in XLA_TEST_F()