You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by jw...@apache.org on 2021/02/25 22:36:21 UTC
[tvm] branch main updated: use checked_type instead of
type_annotation (#7522)
This is an automated email from the ASF dual-hosted git repository.
jwfromm pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 6274a7f use checked_type instead of type_annotation (#7522)
6274a7f is described below
commit 6274a7f56c8966391b643e4c5f9377154205ae62
Author: Matthew Brookhart <mb...@octoml.ai>
AuthorDate: Thu Feb 25 15:36:05 2021 -0700
use checked_type instead of type_annotation (#7522)
---
python/tvm/relay/frontend/onnx.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py
index 1e5dad4..58c2dbc 100644
--- a/python/tvm/relay/frontend/onnx.py
+++ b/python/tvm/relay/frontend/onnx.py
@@ -1753,7 +1753,7 @@ class LSTM(RNN):
P = inputs[7]
num_directions = infer_shape(W)[0]
- W_dtype = infer_type(W).type_annotation.dtype
+ W_dtype = infer_type(W).checked_type.dtype
if num_directions != 1:
raise NotImplementedError("Bidirectional LSTMs not yet supported.")
@@ -1865,7 +1865,7 @@ class GRU(RNN):
linear_before_reset = attr.get("linear_before_reset", 0)
num_directions = infer_shape(W)[0]
- W_dtype = infer_type(W).type_annotation.dtype
+ W_dtype = infer_type(W).checked_type.dtype
if num_directions != 1:
raise NotImplementedError("Bidirectional GRUs not yet supported.")