-
Honglin Yuan authored
Remove `@tf.function` decorator for `predict_on_batch` to workaround a tf.data performance regression. PiperOrigin-RevId: 390692772
aa1bd1f0
Remove `@tf.function` decorator for `predict_on_batch` to workaround a tf.data performance regression. PiperOrigin-RevId: 390692772