You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ma...@apache.org on 2023/04/24 22:16:05 UTC
[tvm] branch main updated: [Torch] fix typo in new_full (#14699)
This is an automated email from the ASF dual-hosted git repository.
masahi 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 1a171396fa [Torch] fix typo in new_full (#14699)
1a171396fa is described below
commit 1a171396fae3a721dd389bab79ea21cac5ab5a1c
Author: liquanfeng <li...@qq.com>
AuthorDate: Tue Apr 25 06:15:57 2023 +0800
[Torch] fix typo in new_full (#14699)
Co-authored-by: liquanfeng <li...@foxmail.com>
---
python/tvm/relay/frontend/pytorch.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py
index 7e6205355a..c81e882526 100644
--- a/python/tvm/relay/frontend/pytorch.py
+++ b/python/tvm/relay/frontend/pytorch.py
@@ -880,7 +880,7 @@ class PyTorchOpConverter:
dtype = _convert_dtype_value(inputs[3])
else:
# if dtype is None, use the dtype of the input tensor
- dtype = self.infer_type(input[0])
+ dtype = self.infer_type(inputs[0])
return self.full_impl(data, fill_value, dtype)