Searched defs:out_grad (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 93 def _SparseTensorDenseAddGrad(op, out_grad): argument
100 def _SparseReduceSumGrad(op, out_grad): argument
|
| D | gradients_test.py | 1164 def Grad(out_grad, variables=None): # pylint: disable=redefined-outer-name argument
1215 def Grad(out_grad, variables=None): # pylint: disable=redefined-outer-name argument
1253 def Grad(out_grad, variables=None): # pylint: disable=redefined-outer-name argument
1277 def Grad(out_grad, variables=None): # pylint: disable=redefined-outer-name argument
1368 def Grad(out_grad, variables=None): # pylint: disable=redefined-outer-name argument
1379 def Grad(out_grad, variables=None): # pylint: disable=redefined-outer-name argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_op_test.py | 296 out_grad, argument
|
| /external/pytorch/torch/csrc/autograd/ |
| D | custom_function.cpp | 121 const auto& out_grad = forward_grads[i]; in _process_forward_mode_AD() local
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesBinaryOps.cpp | 300 …Tensor out_grad = maybePadToLogicalRank(moveBatchDimToFront(grad, grad_bdim), grad_bdim, out_logic… in log_sigmoid_backward_batch_rule() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2778 Value out_grad = op.grad(); in matchAndRewrite() local
|