diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index 15383af47500c..297bc538023a7 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -897,7 +897,7 @@ def _impl_v9(cls, inputs, attr, params): scale_h = scales[-2] scale_w = scales[-1] scale_d = scales[-3] - + layout = 'NCDHW' out = _op.nn.upsampling3d(inputs[0], scale_d,