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 <mbrookh...@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.")