Searched refs:implicit_grad (Results 1 – 12 of 12) sorted by relevance
73 return backprop.implicit_grad(inner)()[0][0]266 return backprop.implicit_grad(inner)()[0][0]326 self.assertAllEqual(backprop.implicit_grad(f)()[0][0], 2.0)335 self.assertAllEqual(backprop.implicit_grad(f)()[0][0], 2.0)337 self.assertAllEqual(backprop.implicit_grad(f)()[0][0], 2.0)402 g = backprop.implicit_grad(g)(constant_op.constant(1.0))[0][0]429 grad_fn = backprop.implicit_grad(sum_gather)431 defun_grad_fn = backprop.implicit_grad(def_function.function(sum_gather))
129 grads_and_vars = backprop.implicit_grad(fn)()279 grad = backprop.implicit_grad(f)()[0][0]310 grads = backprop.implicit_grad(f)()647 self.assertEqual(backprop.implicit_grad(f)()[0][0].cpu().numpy(), 1.0)694 self.assertEqual(backprop.implicit_grad(f)()[0][0], None)1177 grad = backprop.implicit_grad(fn)()[0][0]1244 g = backprop.implicit_grad(f)
259 def implicit_grad(f): function
99 gradients_fn = backprop.implicit_grad(self._loss_fn)
155 grad_fn = backprop.implicit_grad(loss)213 grad_fn = backprop.implicit_grad(loss)
749 grad = backprop.implicit_grad(loss)(self.v)760 grad = backprop.implicit_grad(lambda v: v**2)(v)779 grad = backprop.implicit_grad(loss)(self.v)794 grad = backprop.implicit_grad(loss)()
78 gradients_fn = backprop.implicit_grad(loss_fn)
139 grad_function = backprop.implicit_grad(scale_matrix)
196 grad_function = backprop.implicit_grad(scale_matrix)
174 grads = backprop.implicit_grad(f)()
2384 grad_function = backprop.implicit_grad(scale_matrix)
2697 grad_function = backprop.implicit_grad(scale_matrix)