Remove legacy sampling ops (stateful) from auto control deps.
This fixes @tf.function's inability to lift basic Variable initialization when there are other stateful ops above the location the variable is used in the function (e.g. tf.Assert). Also add better error information about which dependencies are preventing auto lifting. PiperOrigin-RevId: 233853786
Loading
Please sign in to comment