Skip to content

Instantly share code, notes, and snippets.

@rahulremanan

rahulremanan/named_tuples--tf_function--04.py

Created February 28, 2023 16:38
Show Gist options
  • Save rahulremanan/9f6267755b04182541067900c9719876 to your computer and use it in GitHub Desktop.
Save rahulremanan/9f6267755b04182541067900c9719876 to your computer and use it in GitHub Desktop.
Download ZIP
Performance optimization through a decorated Tensorflow function
Raw
named_tuples--tf_function--04.py
@tf.function
def run_named_tuple_graph():
_ijk0 = (tf.constant(0), Pair(tf.constant(1), tf.constant(2)))
_ijk_fin = tf.while_loop(c, b, _ijk0)
return _ijk_fin
ijk_final = run_named_tuple_graph()
print(ijk_final[-1].j.numpy(), ijk_final[-1].k.numpy())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment