Searched defs:grad_op (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | manip_grad.cc | 30 auto grad_op = Roll(scope, grad_inputs[0], Neg(scope, shift), axis); in RollGrad() local
|
| D | array_grad.cc | 50 auto grad_op = Unstack(scope, grad_inputs[0], N, Unstack::Axis(axis)); in PackGrad() local
|
| /external/pytorch/test/functorch/ |
| D | test_ops.py | 2529 def test_view_then_inplace(self, device, dtype, op, grad_op): argument
2566 def test_view_then_inplace_list_return(self, device, dtype, op, grad_op): argument
2592 def test_view_then_inplace_special(self, grad_op): argument
|