Searched defs:primals (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop.py | 332 def __init__(self, primals, tangents): argument
386 def _watch(self, primals, tangents): argument
417 def jvp(self, primals, unconnected_gradients=UnconnectedGradients.NONE): argument
453 def _batch_accumulator(cls, primals, tangents): argument
|
| D | forwardprop_test.py | 59 def _jvp(f, primals, tangents): argument
67 def _jacfwd(f, primals): argument
97 def _jvp_batch_matmul(f, primals, tangent_batch): argument
144 def _hvp(f, primals, tangents): argument
200 primals, argument
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | forwardprop_test.py | 27 def _jvp(f, primals, tangents): argument
35 def _jacfwd(f, primals): argument
72 def _hvp(f, primals, tangents): argument
128 primals, argument
|