Skip to content

Instantly share code, notes, and snippets.

@vinx13

vinx13/partial_eval_bug2.py

Created August 10, 2019 00:29
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 vinx13/1bb2d83543511962ba2ac3c030ec4eb4 to your computer and use it in GitHub Desktop.
Save vinx13/1bb2d83543511962ba2ac3c030ec4eb4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
partial_eval_bug2.py
import tvm
import tvm.relay as relay
x = relay.var("x", shape=(16,), dtype='float32')
y = relay.var("y", shape=(16,), dtype='float32')
c = relay.Let(x, y, x)
f = relay.Function([y], relay.expr.Tuple([c, c]))
mod = relay.Module.from_expr(f)
mod = relay.transform.PartialEvaluate()(mod)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment