Searched refs:_apply_gradients_cross_replica (Results 1 – 1 of 1) sorted by relevance
711 self._apply_gradients_cross_replica,714 def _apply_gradients_cross_replica(self, distribution, grads_and_vars, name, member in LossScaleOptimizer