Skip to content

Instantly share code, notes, and snippets.

@jlebar

jlebar/unsupported_op.py

Created November 7, 2018 01:21
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jlebar/3de338daacbb5bfcf19265e0356cc89a to your computer and use it in GitHub Desktop.
Save jlebar/3de338daacbb5bfcf19265e0356cc89a to your computer and use it in GitHub Desktop.
Download ZIP
Example of TF op that's not supported by XLA
Raw
unsupported_op.py
def model_fn_unsupported_op(x):
return tf.where(tf.cast(x, tf.bool))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment