We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d24634a commit f196a81Copy full SHA for f196a81
tests/python/contrib/test_coreml_codegen.py
@@ -76,12 +76,14 @@ def _create_graph_annotated():
76
func2 = func2.with_attr("global_symbol", target + "_2")
77
gv2 = relay.GlobalVar(target + "_2")
78
mod[gv2] = func2
79
+ mod = relay.transform.InferType()(mod)
80
81
# body
82
x = relay.var("x", shape=shape)
83
y = relay.var("y", shape=shape)
84
func = relay.Function([x, y], gv0(y) - gv2(x))
85
mod["main"] = func
86
87
88
return mod
89
0 commit comments