Searched refs:_jvp (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | forwardprop_test.py | 27 def _jvp(f, primals, tangents): function
50 _jvp(f, primals, tf.nest.pack_sequence_as(primals,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop_test.py | 59 def _jvp(f, primals, tangents): function
83 _jvp(f, primals, nest.pack_sequence_as(primals,
94 functools.partial(_jvp, tf_function, primal), tangents)
410 primal, tangent = _jvp(math_ops.sin, (constant_op.constant(0.1),),
|