From 8296d1459ca845b1259195d47c5977a0d0352dcd Mon Sep 17 00:00:00 2001 From: masahi Date: Mon, 16 Dec 2019 16:15:03 +0900 Subject: [PATCH] fix onnx shape dtype --- python/tvm/relay/frontend/onnx.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index e4e1e9e2616a..07e69564dde9 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -687,8 +687,7 @@ class Shape(OnnxOpConverter): @classmethod def _impl_v1(cls, inputs, attr, params): - # TODO(@jroesch): use shape_of once it has been fixed) - return _op.shape_of(inputs[0]) + return _op.shape_of(inputs[0], "int64") class Cast(OnnxOpConverter): """ Operator converter for Cast.