Searched refs:set_use_auto_spmd_partitioning (Results 1 – 9 of 9) sorted by relevance
80 ExecutableBuildOptions& ExecutableBuildOptions::set_use_auto_spmd_partitioning( in set_use_auto_spmd_partitioning() function in xla::ExecutableBuildOptions118 output.set_use_auto_spmd_partitioning(use_auto_spmd_partitioning()); in ToProto()147 output.set_use_auto_spmd_partitioning(input.use_auto_spmd_partitioning()); in ExecutableBuildOptionsFromProto()199 execution_options.set_use_auto_spmd_partitioning( in CreateExecutionOptions()
98 ExecutableBuildOptions& set_use_auto_spmd_partitioning(
341 proto.set_use_auto_spmd_partitioning(use_auto_spmd_partitioning_); in ToProto()507 module->set_use_auto_spmd_partitioning(proto.use_auto_spmd_partitioning()); in CreateFromProto()538 module_config.set_use_auto_spmd_partitioning( in CreateModuleConfigFromShape()
92 execution_options.set_use_auto_spmd_partitioning( in CompileAheadOfTime()
100 config->set_use_auto_spmd_partitioning( in CreateModuleConfig()
158 void set_use_auto_spmd_partitioning(bool use_auto_spmd_partitioning) { in set_use_auto_spmd_partitioning() function
156 void set_use_auto_spmd_partitioning(bool use) { in set_use_auto_spmd_partitioning() function
507 config.set_use_auto_spmd_partitioning(c_config.use_auto_spmd_partitioning); in FromC()
722 &ExecutableBuildOptions::set_use_auto_spmd_partitioning) in BuildXlaCompilerSubmodule()