Tf.print with XLA compilation?

Hello,

Does anyone know whether tf.print (or an equivalent workaround) can be made to work with XLA compilation? I get an error saying that XLA does not recognize the “printV2” operation if I have a tf.print statement inside a function decorated with @tf.function(jit_compile=True).

Thanks!

Do you have already a github ticket?

yes, thank you very much!