Skip to content

T
tic
Unverified Commit ea07f740 by Tianqi Chen Committed by GitHub
Options

Update test_op_level1.py

parent 710af087
Showing with 1 additions and 1 deletions
tests/python/relay/test_op_level1.py
......@@ -25,7 +25,7 @@ def test_softmax():
ib.ret(func)
func = relay.ir_pass.infer_type(ib.env, func.to_func())
ftype = func.checked_type()
ftype = func.checked_type
assert ftype.ret_type == relay.ty.TensorType((n, d), "float32")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment