This is an automated email from the ASF dual-hosted git repository.
mshr 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 899e1210a6 fixing incorrect docstring in upsampling.py (#17845)
899e1210a6 is described below
commit 899e1210a605fff63bf3924a4ce107a540345839
Author: Cookiee235 <[email protected]>
AuthorDate: Thu Apr 17 17:14:59 2025 +0800
fixing incorrect docstring in upsampling.py (#17845)
Update upsampling.py
fix the incorrect docstring
---
python/tvm/topi/nn/upsampling.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/python/tvm/topi/nn/upsampling.py b/python/tvm/topi/nn/upsampling.py
index e1661bd9be..35a488d4a4 100644
--- a/python/tvm/topi/nn/upsampling.py
+++ b/python/tvm/topi/nn/upsampling.py
@@ -34,7 +34,7 @@ def upsampling(
Parameters
----------
- inputs : tvm.te.Tensor
+ data : tvm.te.Tensor
inputs is a 4-D tensor with shape
[batch, channel, in_height, in_width]
or [batch, in_height, in_width, channel]
@@ -120,7 +120,7 @@ def upsampling3d(
Parameters
----------
- inputs : tvm.te.Tensor
+ data : tvm.te.Tensor
inputs is a 5-D tensor with shape
[batch, channel, in_depth, in_height, in_width]
or [batch, in_depth, in_height, in_width, channel]