from tensorflow.python.util.tf_export import tf_export

Input:

Output:

Prompt: