The example code is like this:
import tvm.relay.op.nn.nn a = tvm.placeholder((1,3,16,16), name='a') b = tvm.relay.op.nn.nn.relu(a)
I get the following error:
Traceback (most recent call last):
File “”, line 1, in
File “/home/pengxihan/tvm/python/tvm/relay/op/nn/nn.py”, line 524, in relu
return _make.relu(data)
File “/home/pengxihan/tvm/python/tvm/_ffi/_ctypes/function.py”, line 209, in call
raise get_last_ffi_error()
tvm._ffi.base.TVMError: Traceback (most recent call last):
[bt] (3) /home/pengxihan/tvm/build/libtvm.so(TVMFuncCall+0x61) [0x7f5740320c41]
[bt] (2) /home/pengxihan/tvm/build/libtvm.so(+0xcce496) [0x7f573ffc4496]
[bt] (1) /home/pengxihan/tvm/build/libtvm.so(+0xb8b304) [0x7f573fe81304]
[bt] (0) /home/pengxihan/tvm/build/libtvm.so(+0x842eb2) [0x7f573fb38eb2]
File “/home/pengxihan/tvm/include/tvm/packed_func_ext.h”, line 141
TVMError: Check failed: NodeTypeChecker: :Check(sptr.get()): Expected type relay.Expr but get Tensor
It seems that relay operator only accept relay.Expr as its input, but I want to give a tvm Tensor as its input and then I want to use tvm.create_schdule function.