Searched refs:_compute_gradients (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | external_optimizer.py | 114 loss_grads = _compute_gradients(loss, self._vars)
116 _compute_gradients(equality, self._vars)
120 _compute_gradients(inequality, self._vars)
442 def _compute_gradients(tensor, var_list): function
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
| D | loss_scale_optimizer.py | 72 def _compute_gradients(self, loss, var_list, grad_loss=None): member in LossScaleOptimizer
74 …grads_and_vars = self._optimizer._compute_gradients(loss, var_list, # pylint: disable=protected-a…
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 297 grads_and_vars = self._compute_gradients(
302 def _compute_gradients(self, loss, var_list, grad_loss=None): member in OptimizerV2
|
| D | optimizer_v2_test.py | 193 grads_and_vars = sgd._compute_gradients(loss, [var0, var1])
232 grads_and_vars = sgd._compute_gradients(f, [x])
|
| D | gradient_descent_test.py | 200 grads_and_vars = opt._compute_gradients(loss, vars_)
|