[Relay][ONNX] KeyError: 'W' when compiling from onnx

Development
#1

The following error occurs when I try to compile an onnx model. From what I can tell using Netron, there is an input W. However, it’s not another node but instead some (static) initialization data for the current node.

  File "/home/martin/Dev/xyz/src/tvm/compile_model.py", line 79, in compile_model
    relay_func, params = relay.frontend.from_onnx(onnx_model, shape=input_shape_dict, dtype=input_dtype_dict)
  File "/home/martin/.local/lib/python3.6/site-packages/tvm-0.6.dev0-py3.6-linux-x86_64.egg/tvm/relay/frontend/onnx.py", line 1126, in from_onnx
    sym, params = g.from_onnx(graph, opset)
  File "/home/martin/.local/lib/python3.6/site-packages/tvm-0.6.dev0-py3.6-linux-x86_64.egg/tvm/relay/frontend/onnx.py", line 942, in from_onnx
    inputs = [self._nodes[self._renames.get(i, i)] for i in node.input]
  File "/home/martin/.local/lib/python3.6/site-packages/tvm-0.6.dev0-py3.6-linux-x86_64.egg/tvm/relay/frontend/onnx.py", line 942, in <listcomp>
    inputs = [self._nodes[self._renames.get(i, i)] for i in node.input]
KeyError: 'W'
#2

Update: From what I know now, it looks like TVM doesn’t take the onnx_model.graph.initializer into account when creating a list of inputs like above.

#3

The model was deleted which is why I uploaded it here: