From a407f2717579954bcd60ba5f13bd45eb51acfcef Mon Sep 17 00:00:00 2001 From: liquanfeng Date: Sat, 22 Apr 2023 20:56:42 +0800 Subject: [PATCH] [Torch] fix typo in new_full --- 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 7e6205355aca..c81e882526de 100644 --- a/python/tvm/relay/frontend/pytorch.py +++ b/python/tvm/relay/frontend/pytorch.py @@ -880,7 +880,7 @@ def new_full(self, inputs, input_types): 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)