[Relay to Onnx Conversion test] Fixed relay.var initialization (apach…

…e#8322) * Fixed issue in bias variable initialization by passing arg name (shape)
ylc · Sep 29, 2021 · 8c25cfc · 8c25cfc
1 parent 5a98a81
commit 8c25cfc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/contrib/test_onnx.py b/tests/python/contrib/test_onnx.py
@@ -89,7 +89,7 @@ def test_bias_add():
         bshape = (2,)
         rtol = 1e-2 if dtype == "float16" else 1e-5
         x = relay.var("x", shape=xshape, dtype=dtype)
-        bias = relay.var("bias", dtype=dtype)
+        bias = relay.var("bias", shape=bshape, dtype=dtype)
         z = relay.nn.bias_add(x, bias)
         func = relay.Function([x, bias], z)