Could you please provide the code for which you are facing error. Thanks!

`
import tensorflow as tf

@tf.function(jit_compile=True)
def print_ten():
for i in range(10):
tf.print('printing')
return

print_ten()
`

Why not

import tensorflow as tf

@tf.function(jit_compile=True)
def print_ten():
  for i in range(10):
    print('printing ' +str(i))
  return

print_ten()

Normal print functions works without any limitations, please find the gist here. Thanks!

The problem is that with normal print, the print statements are only executed on the first call to the function if the function is called multiple times with the same arguments (since tf will use an already made graph, which does not include the calls to print) - (please see https://www.tensorflow.org/guide/function under section "what is tracing?" for details)

tf.print() on the other hand will always print in graph mode (without XLA), even on non-first calls. However, tf.print() is not currently supported when XLA compilation is used.

Hope this clarifies the nature of the problem -please do let me know if you need more info/code examples of the issue.

For example,

`import tensorflow as tf

@tf.function(jit_compile=True)
def print_ten():
for i in range(10):
print('printing ' +str(i))
return

print_ten()
print_ten()
`
Will only print 10 times (corresponding to the first call). If the regular print is replaced with tf.print(), the code will NOT run since XLA does not support tf.print at this time.

Yes if you want to print on every function call it is not supported. You can print the returned value outside of the compiled function.
I don't know if it is still updated but here you can find the list of the supported ops:
#14798 (comment)

Yup - that's why I made a feature request for adding tf.print() to the supported ops for XLA compilation.

Is there a technical reason why tf.print is not supported using XLA?
I am really missing this, especially for custom training loops.

@sachinprasadhs any updates on this please? tf.print does not work with xla

Detected unsupported operations when trying to compile graph __inference_run_step_2702[] on XLA_CPU_JIT: PrintV2 (No registered 'PrintV2' OpKernel for XLA_CPU_JIT devices compatible with node

@MIHIRKHAMBETE !
I am able to replicate this issue with 2.11 version.
Thank you!

@sachinprasadhs Please help forward this to XLA or ML API team. Debugging graph mode issues often relies on tf.print, since XLA graph cannot do tf.print, it's extremely hard to debug when I see bugs specific to XLA graph. Thanks!