Searched refs:use_spmd_partitioning (Results 1 – 21 of 21) sorted by relevance
75 bool use_spmd_partitioning) { in set_use_spmd_partitioning() argument76 use_spmd_partitioning_ = use_spmd_partitioning; in set_use_spmd_partitioning()117 output.set_use_spmd_partitioning(use_spmd_partitioning()); in ToProto()146 output.set_use_spmd_partitioning(input.use_spmd_partitioning()); in ExecutableBuildOptionsFromProto()198 build_options.use_spmd_partitioning()); in CreateExecutionOptions()
91 bool use_spmd_partitioning() const { return use_spmd_partitioning_; } in use_spmd_partitioning() function92 ExecutableBuildOptions& set_use_spmd_partitioning(bool use_spmd_partitioning);
153 void set_use_spmd_partitioning(bool use_spmd_partitioning) { in set_use_spmd_partitioning() argument154 use_spmd_partitioning_ = use_spmd_partitioning; in set_use_spmd_partitioning()156 bool use_spmd_partitioning() const { return use_spmd_partitioning_; } in use_spmd_partitioning() function
91 execution_options.set_use_spmd_partitioning(options.use_spmd_partitioning()); in CompileAheadOfTime()
97 execution_options->use_spmd_partitioning()); in CreateModuleConfig()
736 !module->config().use_spmd_partitioning()) { in Run()748 if (module->config().use_spmd_partitioning() && in Run()
92 virtual bool use_spmd_partitioning() const { return false; } in use_spmd_partitioning() function
275 !ar->GetModule()->config().use_spmd_partitioning())) { in MatchReduceScatter()
537 execution_options->use_spmd_partitioning()); in CreateModuleConfigFromShape()
500 parent()->config().use_spmd_partitioning())) { in ComputeChannelDependencies()
460 hlo_config.use_spmd_partitioning = config.use_spmd_partitioning(); in ToC()506 config.set_use_spmd_partitioning(c_config.use_spmd_partitioning); in FromC()
308 bool use_spmd_partitioning; member
38 bool use_spmd_partitioning = 6; field
202 c_options.use_spmd_partitioning = in ConvertCppCompileOptionsToCCompileOptions()203 options.executable_build_options.use_spmd_partitioning(); in ConvertCppCompileOptionsToCCompileOptions()
326 bool use_spmd_partitioning; member
212 c_option->use_spmd_partitioning); in ConvertCCompileOptionstoCppCompileOptions()
251 use_spmd_partitioning: bool
718 &ExecutableBuildOptions::use_spmd_partitioning, in BuildXlaCompilerSubmodule()
434 if (module->config().use_spmd_partitioning()) { in RunHloPassesThroughLayoutAssn()
491 bool use_spmd_partitioning = 11; field
353 if (hlo_module->config().use_spmd_partitioning()) { in OptimizeHloModule()