Wraps fwd in a defun with custom gradient bak and calls it with args. Args: fwd: A callable xs: Nested Structure -> ys: Nested Structure. args: A Nested Structure of tf.Tensor or None. bak: A callable xs, ys, dys: Nested Structure -> dxs[, dcapture]: Nested Structure. The custom
(fwd, args=None, bak=None, bak_as_function=False, device=None)
source not stored for this graph (policy: none)
no test coverage detected
searching dependent graphs…