Searched defs:primals (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop.py | 328 def __init__(self, primals, tangents): argument
382 def _watch(self, primals, tangents): argument
413 def jvp(self, primals, unconnected_gradients=UnconnectedGradients.NONE): argument
449 def _batch_accumulator(cls, primals, tangents): argument
|
| D | forwardprop_test.py | 56 def _jvp(f, primals, tangents): argument
64 def _jacfwd(f, primals): argument
98 def _jvp_batch_matmul(f, primals, tangent_batch): argument
145 def _hvp(f, primals, tangents): argument
201 primals, argument
|