Searched refs:get_gradient_function (Results 1 – 4 of 4) sorted by relevance
343 self.assertEqual(None, ops.get_gradient_function(x.op))344 self.assertEqual(None, ops.get_gradient_function(y.op))
673 grad_fn = ops.get_gradient_function(op)
1738 fn = ops.get_gradient_function(y.op)1747 fn = ops.get_gradient_function(y.op)1757 ops.get_gradient_function(y.op)
2704 def get_gradient_function(op): function