When compiling following model with TVM, it will error.
Traceback (most recent call last):
File "check.py", line 11, in <module>
mod, params = relay.frontend.from_onnx(onnx_model, {})
File "/Users/luyaor/Documents/tvm/python/tvm/relay/frontend/onnx.py", line 2806, in from_onnx
mod, params = g.from_onnx(graph, opset, freeze_params)
File "/Users/luyaor/Documents/tvm/python/tvm/relay/frontend/onnx.py", line 2613, in from_onnx
op = self._convert_operator(op_name, inputs, attr, opset)
File "/Users/luyaor/Documents/tvm/python/tvm/relay/frontend/onnx.py", line 2721, in _convert_operator
sym = convert_map[op_name](inputs, attrs, self._params)
File "/Users/luyaor/Documents/tvm/python/tvm/relay/frontend/onnx.py", line 820, in _impl_v1
input_channels = infer_shape(inputs[0])[1]
IndexError: tuple index out of range
Description
When compiling following model with TVM, it will error.
The model(with ONNX as frontend) with error is as follows, check bug.onnx in bug3.zip
Error Log
How to reproduce
Environment
Python3, with tvm, onnx
tvm version:
c31e338Wed Dec 9 14:52:58 2020 +0900Download bug3.zip
Run
python check.py.