Searched refs:use_auto_spmd_partitioning (Results 1 – 16 of 16) sorted by relevance
81 bool use_auto_spmd_partitioning) { in set_use_auto_spmd_partitioning() argument82 use_auto_spmd_partitioning_ = use_auto_spmd_partitioning; in set_use_auto_spmd_partitioning()118 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()200 build_options.use_auto_spmd_partitioning()); in CreateExecutionOptions()
95 bool use_auto_spmd_partitioning() const { in use_auto_spmd_partitioning() function99 bool use_auto_spmd_partitioning);
158 void set_use_auto_spmd_partitioning(bool use_auto_spmd_partitioning) { in set_use_auto_spmd_partitioning() argument159 use_auto_spmd_partitioning_ = use_auto_spmd_partitioning; in set_use_auto_spmd_partitioning()160 if (use_auto_spmd_partitioning) { in set_use_auto_spmd_partitioning()171 bool use_auto_spmd_partitioning() const { in use_auto_spmd_partitioning() function
93 options.use_auto_spmd_partitioning()); in CompileAheadOfTime()
101 execution_options->use_auto_spmd_partitioning()); in CreateModuleConfig()
93 virtual bool use_auto_spmd_partitioning() const { return false; } in use_auto_spmd_partitioning() function
507 module->set_use_auto_spmd_partitioning(proto.use_auto_spmd_partitioning()); in CreateFromProto()539 execution_options->use_auto_spmd_partitioning()); in CreateModuleConfigFromShape()
160 bool use_auto_spmd_partitioning() const { in use_auto_spmd_partitioning() function
495 bool use_auto_spmd_partitioning = 16; field
127 // use_auto_spmd_partitioning=true.131 // use_auto_spmd_partitioning=true.
461 hlo_config.use_auto_spmd_partitioning = config.use_auto_spmd_partitioning(); in ToC()507 config.set_use_auto_spmd_partitioning(c_config.use_auto_spmd_partitioning); in FromC()
309 bool use_auto_spmd_partitioning; member
41 bool use_auto_spmd_partitioning = 7; field
494 bool use_auto_spmd_partitioning = 15; field497 // use_auto_spmd_partitioning=true.501 // use_auto_spmd_partitioning=true.
252 use_auto_spmd_partitioning: bool
721 &ExecutableBuildOptions::use_auto_spmd_partitioning, in BuildXlaCompilerSubmodule()