Searched defs:grad_loss (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 241 def _CTCLossGradImpl(op, grad_loss, _): argument
260 def _CTCLossGrad(op, grad_loss, _): argument
275 def _CTCLossV2Grad(op, grad_loss, _): argument
690 def _ctc_loss_grad(op, grad_loss, _): argument
1055 def grad(grad_loss): argument
|
| D | nn_grad.py | 519 def _SoftmaxCrossEntropyWithLogitsGrad(op, grad_loss, grad_grad): argument
544 def _SparseSoftmaxCrossEntropyWithLogitsGrad(op, grad_loss, grad_grad): argument
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 474 def _get_gradients(self, tape, loss, var_list, grad_loss=None): argument
500 def minimize(self, loss, var_list, grad_loss=None, name=None, tape=None): argument
535 def _compute_gradients(self, loss, var_list, grad_loss=None, tape=None): argument
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 90 grad_loss=None): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 366 grad_loss=None): argument
427 grad_loss=None): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | optimizer_v1.py | 766 def minimize(self, loss, var_list, grad_loss=None, tape=None): argument
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad.cc | 90 auto grad_loss = grad_inputs[0]; in SoftmaxCrossEntropyWithLogitsGrad() local
|
| /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): argument
|