Searched refs:_hvp (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | forwardprop_test.py | 72 def _hvp(f, primals, tangents): function
120 functools.partial(_hvp, f, params),
256 forward_over_back_hvp, = _hvp(variable_input_fn,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 144 def _hvp(f, primals, tangents): function
192 functools.partial(_hvp, f, params),
660 _hvp(fun, (primals,), (tangents,))
670 forwardback_hvp_eager, = _hvp(fun, (primals,), (tangents,))
671 forwardback_hvp_function, = def_function.function(_hvp)(fun, (primals,),
|