Fix issue in which the default learning phase was incorrectly captured when
using a tf.function in a subclassed Layer or Model. PiperOrigin-RevId: 234088875
Loading
Please sign in to comment
using a tf.function in a subclassed Layer or Model. PiperOrigin-RevId: 234088875