Searched refs:_apply_gradients (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 220 return self._apply_gradients(distribution, grads_and_vars, global_step,
228 def _apply_gradients(self, distribution, grads_and_vars, global_step, name): member in MixedPrecisionLossScaleOptimizer
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 731 self._apply_gradients,
748 def _apply_gradients(self, grads, wrapped_vars, name, member in LossScaleOptimizer
|