search

jiazhihao / TASO

The Tensor Algebra SuperOptimizer for Deep Learning
Apache License 2.0
693 stars 90 forks source link

Bertsquad-8 from onnx-models: AssertionError: Slice requires at least 3 inputs #66

Open sergei-mironov opened 4 years ago

sergei-mironov commented 4 years ago

Hi. Just found an error when applied TASO to bertsquad-8 model of onnx-models. Note, that I was on dce8c4d967f at the time of experiment

Traceback (most recent call last):
  File "taso/examples/test_onnx.py", line 12, in <module>
    graph = taso.load_onnx(args.file)
  File "/opt/conda/lib/python3.8/site-packages/taso-0.1.0-py3.8-linux-x86_64.egg/taso/__init__.py", line 805, in load_onnx
    outputs = xf_operators[op.op_type](op, graph, tensors, model.graph.initializer)
  File "/opt/conda/lib/python3.8/site-packages/taso-0.1.0-py3.8-linux-x86_64.egg/taso/__init__.py", line 539, in _slice
    assert len(inputs) >= 3, "Slice requires at least 3 inputs"
AssertionError: Slice requires at least 3 inputs