tf.autograph.to_code

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

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

Used in the notebooks

Used in the guide
Better performance with tf.function Introduction to graphs and tf.function

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.

Note: If a function has been decorated with tf.function, pass its underlying Python function, rather than the callable that `tf.function creates:

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

Args
`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.

Returns
The converted code as string.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/autograph/to_code