Searched refs:enable_automatic_model_parallelism (Results 1 – 3 of 3) sorted by relevance
113 bool enable_automatic_model_parallelism = 14; field
135 bool enable_automatic_model_parallelism, bool enable_xla_param_broadcast);
4546 bool enable_automatic_model_parallelism, bool enable_xla_param_broadcast) { in SetDistributedTpuRewritePassOptions() argument4553 enable_automatic_model_parallelism_ = enable_automatic_model_parallelism; in SetDistributedTpuRewritePassOptions()