diff --git a/onnxruntime/core/providers/cpu/tensor/upsamplebase.h b/onnxruntime/core/providers/cpu/tensor/upsamplebase.h index a0e7ca1084fef..749b03cb0585a 100644 --- a/onnxruntime/core/providers/cpu/tensor/upsamplebase.h +++ b/onnxruntime/core/providers/cpu/tensor/upsamplebase.h @@ -76,7 +76,7 @@ class UpsampleBase { } auto input_count = info.GetInputCount(); - if (input_count == 1) { // opset < 10 + if (opset < 10) { ORT_THROW_IF_ERROR(info.GetAttrs("scales", scales_)); ORT_THROW_IF_ERROR(ScalesValidation(scales_, mode_)); scales_cached_ = true;