autograph.py

@tf.function
def f(x):
  for i in range(10):  # Static python loop, we'll not convert it
    do_stuff()
  for i in tf.range(10):  # depends on a tensor, we'll convert it

 # function call
 f(10)

  # If you're curious you can inspect the code autograph generates.
  # PS: Read the code at your own risk !!!
  print(tf.autograph.to_code(f))