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!