Searched refs:get_gradient_function (Results 1 – 5 of 5) sorted by relevance
384 self.assertEqual(None, ops.get_gradient_function(x.op))385 self.assertEqual(None, ops.get_gradient_function(y.op))
608 grad_fn = ops.get_gradient_function(op)
2244 fn = ops.get_gradient_function(y.op)2253 fn = ops.get_gradient_function(y.op)2263 ops.get_gradient_function(y.op)
2706 def get_gradient_function(op): function
784 def get_gradient_function(self): member in _DelayedRewriteGradientFunctions