tf.autograph.to_code

TensorFlow 1 version View source on GitHub

Returns the source code generated by AutoGraph, as a string.

tf.autograph.to_code(
    entity, recursive=True, experimental_optional_features=None
)

Example usage:

def f(x):
  if x < 0:
    x = -x
  return x
tf.autograph.to_code(f)
"...def tf__f(x):..."

Also see: tf.autograph.to_graph.

@tf.function
def f(x):
  if x < 0:
    x = -x
  return x
tf.autograph.to_code(f.python_function)
"...def tf__f(x):..."

entity Python callable or class to convert.
recursive Whether to recursively convert any functions that the converted function may call.
experimental_optional_features None, a tuple of, or a single tf.autograph.experimental.Feature value.

The converted code as string.