Searched refs:set_use_spmd_partitioning (Results 1 – 15 of 15) sorted by relevance
74 ExecutableBuildOptions& ExecutableBuildOptions::set_use_spmd_partitioning( in set_use_spmd_partitioning() function in xla::ExecutableBuildOptions117 output.set_use_spmd_partitioning(use_spmd_partitioning()); in ToProto()146 output.set_use_spmd_partitioning(input.use_spmd_partitioning()); in ExecutableBuildOptionsFromProto()197 execution_options.set_use_spmd_partitioning( in CreateExecutionOptions()
92 ExecutableBuildOptions& set_use_spmd_partitioning(bool use_spmd_partitioning);
49 config.set_use_spmd_partitioning(true); in TEST_F()72 config.set_use_spmd_partitioning(true); in TEST_F()
43 config.set_use_spmd_partitioning(true); in TEST_F()68 config.set_use_spmd_partitioning(true); in TEST_F()
42 config.set_use_spmd_partitioning(true); in TEST_F()
46 config.set_use_spmd_partitioning(num_partitions > 1); in RunPass()
153 void set_use_spmd_partitioning(bool use_spmd_partitioning) { in set_use_spmd_partitioning() function168 set_use_spmd_partitioning(true); in set_use_auto_spmd_partitioning()
803 module->config().set_use_spmd_partitioning(true); in TEST_F()906 module->config().set_use_spmd_partitioning(true); in TEST_F()
91 execution_options.set_use_spmd_partitioning(options.use_spmd_partitioning()); in CompileAheadOfTime()
96 config->set_use_spmd_partitioning( in CreateModuleConfig()
203 module->config().set_use_spmd_partitioning(false); in TEST_F()
536 module_config.set_use_spmd_partitioning( in CreateModuleConfigFromShape()
211 ret.executable_build_options.set_use_spmd_partitioning( in ConvertCCompileOptionstoCppCompileOptions()
506 config.set_use_spmd_partitioning(c_config.use_spmd_partitioning); in FromC()
719 &ExecutableBuildOptions::set_use_spmd_partitioning) in BuildXlaCompilerSubmodule()