Searched refs:allow_spmd_sharding_propagation_to_output (Results 1 – 15 of 15) sorted by relevance
146 bool allow_spmd_sharding_propagation_to_output() const { in allow_spmd_sharding_propagation_to_output() function157 bool allow_spmd_sharding_propagation_to_output) { in set_allow_spmd_sharding_propagation_to_output() argument159 allow_spmd_sharding_propagation_to_output; in set_allow_spmd_sharding_propagation_to_output()
127 allow_spmd_sharding_propagation_to_output()); in ToProto()158 input.allow_spmd_sharding_propagation_to_output()); in ExecutableBuildOptionsFromProto()209 build_options.allow_spmd_sharding_propagation_to_output()); in CreateExecutionOptions()
313 bool allow_spmd_sharding_propagation_to_output) { in set_allow_spmd_sharding_propagation_to_output() argument315 allow_spmd_sharding_propagation_to_output; in set_allow_spmd_sharding_propagation_to_output()317 bool allow_spmd_sharding_propagation_to_output() const { in allow_spmd_sharding_propagation_to_output() function
56 bool allow_spmd_sharding_propagation_to_output = false,62 allow_spmd_sharding_propagation_to_output), in is_spmd_()
99 execution_options->allow_spmd_sharding_propagation_to_output()); in CreateModuleConfig()
274 if (config_.allow_spmd_sharding_propagation_to_output()) { in ToCord()552 execution_options->allow_spmd_sharding_propagation_to_output()); in CreateModuleConfigFromShape()
324 bool allow_spmd_sharding_propagation_to_output, in SupportSpatialPartitioning() argument332 !(is_entry_root && allow_spmd_sharding_propagation_to_output)) { in SupportSpatialPartitioning()
70 bool allow_spmd_sharding_propagation_to_output = 12; field
204 c_options.allow_spmd_sharding_propagation_to_output = in ConvertCppCompileOptionsToCCompileOptions()206 .allow_spmd_sharding_propagation_to_output(); in ConvertCppCompileOptionsToCCompileOptions()
328 bool allow_spmd_sharding_propagation_to_output; member
214 c_option->allow_spmd_sharding_propagation_to_output); in ConvertCCompileOptionstoCppCompileOptions()
733 &ExecutableBuildOptions::allow_spmd_sharding_propagation_to_output, in BuildXlaCompilerSubmodule()
447 module->config().allow_spmd_sharding_propagation_to_output()); in RunHloPassesThroughLayoutAssn()
388 hlo_module->config().allow_spmd_sharding_propagation_to_output()); in OptimizeHloModule()
517 bool allow_spmd_sharding_propagation_to_output = 14; field