Skip to content

T
tic
Commit 093405d8 by masahi Committed by Tianqi Chen
Options

fix onnx shape dtype (#4528)

parent 8e3b5d39
Showing with 1 additions and 2 deletions
python/tvm/relay/frontend/onnx.py
...@@ -687,8 +687,7 @@ class Shape(OnnxOpConverter): ...@@ -687,8 +687,7 @@ class Shape(OnnxOpConverter):
@classmethod @classmethod
def _impl_v1(cls, inputs, attr, params): def _impl_v1(cls, inputs, attr, params):
# TODO(@jroesch): use shape_of once it has been fixed) return _op.shape_of(inputs[0], "int64")
return _op.shape_of(inputs[0])
class Cast(OnnxOpConverter): class Cast(OnnxOpConverter):
""" Operator converter for Cast. """ Operator converter for Cast.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment