Home
last modified time | relevance | path

Searched refs:val_and_grad_function (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dbackprop.py412 _, grad = val_and_grad_function(f, params=params)(*args, **kwds)
450 def val_and_grad_function(f, params=None): function
H A Dtape_test.py121 val, (da,) = backprop.val_and_grad_function(forward, ['a'])(aa, bb)
H A Dbackprop_test.py1095 val_and_grads_fn = backprop.val_and_grad_function(fn)
1110 val_and_grad_fn = backprop.val_and_grad_function(fn, params=[1])
1404 val_and_grads_fn = backprop.val_and_grad_function(fn)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/distributions/
H A Dspecial_math_test.py69 v, g = tfe_backprop.val_and_grad_function(fn)(args)