[tvm] branch main updated: [Relay to Onnx Conversion test] Fixed relay.var initialization (#8322)

[ma...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 5177729  [Relay to Onnx Conversion test] Fixed relay.var initialization (#8322)
5177729 is described below

commit 5177729ce3f20b0073c334403136af47b163bb55
Author: schilkunda-amba <65...@users.noreply.github.com>
AuthorDate: Fri Jun 25 13:24:30 2021 -0700

    [Relay to Onnx Conversion test] Fixed relay.var initialization (#8322)
    
    * Fixed issue in bias variable initialization by passing arg name (shape)
---
 tests/python/contrib/test_onnx.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/python/contrib/test_onnx.py b/tests/python/contrib/test_onnx.py
index d99946d..b6723e9 100644
--- 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)