Searched defs:set_optimizer_weight_shard_size (Results 1 – 2 of 2) sorted by relevance
145 void ParallelContext::set_optimizer_weight_shard_size(int64_t optimizer_weight_shard_size) { in set_optimizer_weight_shard_size() function in mindspore::parallel::ParallelContext
549 def set_optimizer_weight_shard_size(self, optimizer_weight_shard_size): member in _AutoParallelContext