Fix fix Defun logic when returning a value captured from an outer scope.
Previously, the following would fail: ```python c = tf.constant(...) @Defun(...) def Foo(...): return c ``` The fix involves ensuring that every output of the function is a member of the function graph, and invoking the capturing logic if it is not. PiperOrigin-RevId: 186705800
Loading
Please sign in to comment