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]