Searched refs:_grad (Results 1 – 7 of 7) sorted by relevance
255 def _grad(op, grad): function329 return _grad337 def _grad(op, grad): function369 return _grad
132 def _grad(f): function151 f = _grad(f)152 traced_f = def_function.function(_grad(traced_f))156 def _grad(f): function177 f = _grad(f)
110 def _grad(f, argnums=0): function212 f=_grad(f),
1765 def _grad(f): function1784 f = _grad(f)
57 def _grad(f, argnums=0): function140 f=_grad(f),
969 def _grad(f): function985 f = _grad(f)
1465 def _grad(self, f, argnums=0): member in VariablesGradientTest1486 f=self._grad(f),