Searched refs:_compute_gradients (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 677 def _compute_gradients(self, loss, var_list, grad_loss=None, tape=None): member in LossScaleOptimizer
681 grads_and_vars = self._optimizer._compute_gradients( # pylint: disable=protected-access
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 531 grads_and_vars = self._compute_gradients(
535 def _compute_gradients(self, loss, var_list, grad_loss=None, tape=None): member in OptimizerV2
|
| D | optimizer_v2_test.py | 207 grads_and_vars = sgd._compute_gradients(loss, [var0, var1])
246 grads_and_vars = sgd._compute_gradients(f, [x])
|
| D | gradient_descent_test.py | 202 grads_and_vars = opt._compute_gradients(loss, vars_)
|