From f4aeaf6c19f9d28c2bf62c82bbf145091e5ab539 Mon Sep 17 00:00:00 2001 From: Cookiee235 Date: Wed, 16 Apr 2025 21:33:38 +0800 Subject: [PATCH] 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 e1661bd9beb5..35a488d4a4ea 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]