Searched refs:set_optimizer_weight_shard_aggregated_save (Results 1 – 5 of 5) sorted by relevance
572 def set_optimizer_weight_shard_aggregated_save(self, optimizer_weight_shard_aggregated_save): member in _AutoParallelContext583 …self._context_handle.set_optimizer_weight_shard_aggregated_save(optimizer_weight_shard_aggregated_…651 …_weight_shard_aggregated_save": auto_parallel_context().set_optimizer_weight_shard_aggregated_save,
62 auto_parallel_context().set_optimizer_weight_shard_aggregated_save(True)
104 void set_optimizer_weight_shard_aggregated_save(bool optimizer_weight_shard_aggregated_save);
149 void ParallelContext::set_optimizer_weight_shard_aggregated_save(bool optimizer_weight_shard_aggreg… in set_optimizer_weight_shard_aggregated_save() function in mindspore::parallel::ParallelContext
195 …imizer_weight_shard_aggregated_save", &ParallelContext::set_optimizer_weight_shard_aggregated_save, in PYBIND11_MODULE()