Home
last modified time | relevance | path

Searched refs:has_entry_computation_layout (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc423 hlo_config.has_entry_computation_layout = in ToC()
424 config.has_entry_computation_layout(); in ToC()
436 if (config.has_entry_computation_layout()) { in ToC()
454 xla::HloModuleConfig config = c_config.has_entry_computation_layout in FromC()
Dc_api_decl.h285 bool has_entry_computation_layout; member
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_config.h80 bool has_entry_computation_layout() const { in has_entry_computation_layout() function
Dhlo_module.cc77 } else if (!config_.has_entry_computation_layout()) { in AddComputationInternal()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc689 const Shape& param_shape = (module_config.has_entry_computation_layout() && in MakeFakeArguments()