tfm.nlp.layers.tf_function_if_eager

Applies the @tf.function decorator only if running in eager mode.