Searched refs:get_gradient_function (Results 1 – 5 of 5) sorted by relevance
382 self.assertEqual(None, ops.get_gradient_function(x.op))383 self.assertEqual(None, ops.get_gradient_function(y.op))
611 grad_fn = ops.get_gradient_function(op)
2295 fn = ops.get_gradient_function(y.op)2304 fn = ops.get_gradient_function(y.op)2314 ops.get_gradient_function(y.op)
2869 def get_gradient_function(op): function
689 def get_gradient_function(self): member in _DelayedRewriteGradientFunctions