Searched refs:set_use_spmd_partitioning (Results 1 – 8 of 8) sorted by relevance
74 ExecutableBuildOptions& ExecutableBuildOptions::set_use_spmd_partitioning( in set_use_spmd_partitioning() function in xla::ExecutableBuildOptions127 execution_options.set_use_spmd_partitioning( in CreateExecutionOptions()
90 ExecutableBuildOptions& set_use_spmd_partitioning(bool use_spmd_partitioning);
94 execution_options.set_use_spmd_partitioning(options.use_spmd_partitioning()); in CompileAheadOfTime()
96 config->set_use_spmd_partitioning( in CreateModuleConfig()
143 void set_use_spmd_partitioning(bool use_spmd_partitioning) { in set_use_spmd_partitioning() function
447 module_config.set_use_spmd_partitioning( in CreateModuleConfigFromShape()
556 &ExecutableBuildOptions::set_use_spmd_partitioning); in BuildXlaCompilerSubmodule()
461 config.set_use_spmd_partitioning(c_config.use_spmd_partitioning); in FromC()