Constructor. Args: fwd: A callable xs: Nested Structure -> ys: Nested Structure. Used to build the TensorFlow graph function that this object encapsulates. fwd_sig: A Nested Structure of tf.TensorSpec representing the input signature of `fwd`, or None (meaning that `
(self,
fwd,
fwd_sig=None,
bak=None,
bak_as_function=False,
device=None)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected