Searched refs:primals (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | forwardprop.py | 328 def __init__(self, primals, tangents): argument
353 for primal in nest.flatten(primals):
360 self._watch(primals, tangents)
382 def _watch(self, primals, tangents): argument
411 nest.map_structure(_watch, primals, tangents)
413 def jvp(self, primals, unconnected_gradients=UnconnectedGradients.NONE): argument
446 return nest.map_structure(_fetch_jvp, primals)
449 def _batch_accumulator(cls, primals, tangents): argument
461 acc = super(ForwardAccumulator, cls).__new__(cls, primals, tangents)
465 for primal, tangent in zip(nest.flatten(primals), nest.flatten(tangents)):
[all …]
|
| D | forwardprop_test.py | 56 def _jvp(f, primals, tangents): argument
58 with forwardprop.ForwardAccumulator(primals, tangents) as acc:
59 primals_out = f(*primals)
64 def _jacfwd(f, primals): argument
67 flat_primals = nest.flatten(primals)
84 _jvp(f, primals, nest.pack_sequence_as(primals,
88 return nest.pack_sequence_as(primals, jac_flat)
98 def _jvp_batch_matmul(f, primals, tangent_batch): argument
100 jac_fwd = _jacfwd(f, primals)
133 primals=[params[argnums]], tangents=param_mask) as acc:
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.autodiff.-forward-accumulator.pbtxt | 7 … argspec: "args=[\'self\', \'primals\', \'tangents\'], varargs=None, keywords=None, defaults=None"
11 …argspec: "args=[\'self\', \'primals\', \'unconnected_gradients\'], varargs=None, keywords=None, de…
|