Searched defs:grad_fn (Results 1 – 11 of 11) sorted by relevance
26 ops::GradFunc grad_fn; in CallGradFunction() local
150 def grad_fn(dy): function167 def grad_fn(dy): function
388 def _MaybeCompile(scope, op, func, grad_fn): argument
74 def grad_fn(_): function
374 ops::GradFunc grad_fn; in IsPrimitiveOpWithNoGrad() local383 ops::GradFunc grad_fn; in CallGradFunction() local
195 def grad_fn(*args, **kwds): function263 def grad_fn(*args, **kwds): function
628 def grad_fn(*output_grads, **kwargs): function631 def grad_fn(*output_grads): function
373 def _BenchmarkGrad(grad_fn, name, device): argument
167 def grad_fn(x): function
505 def grad_fn(x): function
47 def get_filtered_grad_fn(grad_fn): argument