Searched refs:get_optimizer_weight_shard_size (Results 1 – 1 of 1) sorted by relevance
567 def get_optimizer_weight_shard_size(self): member in _AutoParallelContext570 return self._context_handle.get_optimizer_weight_shard_size()674 "optimizer_weight_shard_size": auto_parallel_context().get_optimizer_weight_shard_size,