Skip to content

Instantly share code, notes, and snippets.

@vinx13
Created August 10, 2019 00:29
Show Gist options
  • Save vinx13/1bb2d83543511962ba2ac3c030ec4eb4 to your computer and use it in GitHub Desktop.
Save vinx13/1bb2d83543511962ba2ac3c030ec4eb4 to your computer and use it in GitHub Desktop.
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