Dear All
I can use VM runtime to do inference with batch size = 1
but I would like to do the dynamic batch size
so I set the batch dimension to tvm.relay.Any()
But got the following error
Output : src/tvm/python/tvm/relay/frontend/pytorch.py", line 4237, in get_relay_ty raise RuntimeError(msg) RuntimeError: Shape for Tensors must be lists of ints
input_shape[0] = tvm.relay.Any()
input_name = "input"
shape_list = [(input_name, (input_shape, dtype))]
mod, params = relay.frontend.from_pytorch(s, shape_list, default_dtype = dtype)