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)