Use opset value instead of inferring based on the number of number in…

…puts because for Resize opset 13 and above the number of inputs is 1 to 4.
microsoft · Dec 29, 2023 · aaba09b · aaba09b
1 parent 780fc36
commit aaba09b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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<float>("scales", scales_));
       ORT_THROW_IF_ERROR(ScalesValidation(scales_, mode_));
       scales_cached_ = true;