Searched defs:grad_loss (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 248 def _CTCLossGradImpl(op, grad_loss, _): argument
267 def _CTCLossGrad(op, grad_loss, _): argument
282 def _CTCLossV2Grad(op, grad_loss, _): argument
736 def _ctc_loss_grad(op, grad_loss, _): argument
1109 def grad(grad_loss): argument
|
| D | nn_grad.py | 513 def _SoftmaxCrossEntropyWithLogitsGrad(op, grad_loss, grad_grad): argument
538 def _SparseSoftmaxCrossEntropyWithLogitsGrad(op, grad_loss, grad_grad): argument
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 465 def _get_gradients(self, tape, loss, var_list, grad_loss=None): argument
502 def minimize(self, loss, var_list, grad_loss=None, name=None, tape=None): argument
537 def _compute_gradients(self, loss, var_list, grad_loss=None, tape=None): argument
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 86 grad_loss=None): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 433 grad_loss=None): argument
494 grad_loss=None): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | optimizer_v1.py | 782 def minimize(self, loss, var_list, grad_loss=None, tape=None): argument
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 569 def _compute_gradients(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
|