Searched refs:get_gradients_mean (Results 1 – 3 of 3) sorted by relevance
134 def get_gradients_mean(self): member in _AutoParallelContext137 return self._context_handle.get_gradients_mean()659 "gradients_mean": auto_parallel_context().get_gradients_mean,
174 return auto_parallel_context().get_gradients_mean()
52 gradients_mean = auto_parallel_context().get_gradients_mean()