Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_util.cc118 config->set_static_device_assignment(*device_assignment); in CreateModuleConfig()
Dhlo_module_config.h185 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment() function
Dcompiler.h103 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment() function
Dhlo_module.cc456 module_config.set_static_device_assignment(*device_assignment); in CreateModuleConfigFromShape()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcollective_ops_test.cc316 config.set_static_device_assignment(device_assn); in XLA_TEST_F()
358 config.set_static_device_assignment(e.device_assn); in XLA_TEST_F()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc118 config->set_static_device_assignment(device_assignment.value()); in CreateModuleConfig()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc466 config.set_static_device_assignment( in FromC()