Searched refs:_aggregate_gradients (Results 1 – 3 of 3) sorted by relevance
864 def _aggregate_gradients(self, grads_and_vars): member in LossScaleOptimizer865 return self._optimizer._aggregate_gradients(grads_and_vars) # pylint: disable=protected-access
483 def _aggregate_gradients(self, grads_and_vars): member in OptimizerV2666 grads_and_vars = self._aggregate_gradients(grads_and_vars)
1468 def _aggregate_gradients(self, grads): member in TrainingTest.test_calling_aggregate_gradient._Optimizer1470 return super(_Optimizer, self)._aggregate_gradients(grads)