Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dc_api_conversions.cc468 hlo_config.has_entry_computation_layout = in ToC()
469 config.has_entry_computation_layout(); in ToC()
481 if (config.has_entry_computation_layout()) { in ToC()
499 xla::HloModuleConfig config = c_config.has_entry_computation_layout in FromC()
H A Dc_api_decl.h315 bool has_entry_computation_layout; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_config.h84 bool has_entry_computation_layout() const { in has_entry_computation_layout() function
H A Dhlo_module.cc85 } else if (!config_.has_entry_computation_layout()) { in AddComputationInternal()
269 if (config_.has_entry_computation_layout()) { in ToCord()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dhlo_test_base.cc457 (module_config_0.has_entry_computation_layout() && in RunAndCompareTwoModules()
469 (module_config_1.has_entry_computation_layout() && in RunAndCompareTwoModules()
H A Dtest_utils.cc746 const Shape& param_shape = (module_config.has_entry_computation_layout() && in MakeFakeArguments()