Searched refs:set_use_spmd_partitioning (Results 1 – 11 of 11) sorted by relevance
74 ExecutableBuildOptions& ExecutableBuildOptions::set_use_spmd_partitioning( in set_use_spmd_partitioning() function in xla::ExecutableBuildOptions121 execution_options.set_use_spmd_partitioning( in CreateExecutionOptions()
90 ExecutableBuildOptions& set_use_spmd_partitioning(bool use_spmd_partitioning);
42 config.set_use_spmd_partitioning(true); in TEST_F()
46 config.set_use_spmd_partitioning(num_partitions > 1); in RunPass()
96 config->set_use_spmd_partitioning( in CreateModuleConfig()
94 execution_options.set_use_spmd_partitioning(options.use_spmd_partitioning()); in CompileAheadOfTime()
145 void set_use_spmd_partitioning(bool use_spmd_partitioning) { in set_use_spmd_partitioning() function
203 module->config().set_use_spmd_partitioning(false); in TEST_F()
456 module_config.set_use_spmd_partitioning( in CreateModuleConfigFromShape()
461 config.set_use_spmd_partitioning(c_config.use_spmd_partitioning); in FromC()
639 &ExecutableBuildOptions::set_use_spmd_partitioning); in BuildXlaCompilerSubmodule()