Fix 2D detection bug (#21128)

yuslepukhin · web-flow · commit b95982e588e2 · 2024-06-21T13:58:21.000-07:00
### Description
Should compare two leading dims for 1.f

### Motivation and Context
Vulnerability scanner
diff --git a/onnxruntime/core/providers/cuda/tensor/upsample.cc b/onnxruntime/core/providers/cuda/tensor/upsample.cc
@@ -230,7 +230,7 @@ Status Upsample<T>::BaseCompute(OpKernelContext* context,
           }
 
           const bool is_2D = X_dims.size() == 2;
-          const bool is_nchw = is_2D ? true : (scales[1] == 1.0f && scales[1] == 1.0f);
+          const bool is_nchw = is_2D ? true : (scales[0] == 1.0f && scales[1] == 1.0f);
 
           ORT_RETURN_IF_NOT(is_nchw,
                             "Resize 'Cubic' mode only supports NCWH layout "

Original file line number	Diff line number	Diff line change
`@@ -230,7 +230,7 @@ Status Upsample<T>::BaseCompute(OpKernelContext* context,`
`230`	`230`	`}`
`231`	`231`
`232`	`232`	`const bool is_2D = X_dims.size() == 2;`
`233`		`- const bool is_nchw = is_2D ? true : (scales[1] == 1.0f && scales[1] == 1.0f);`
	`233`	`+ const bool is_nchw = is_2D ? true : (scales[0] == 1.0f && scales[1] == 1.0f);`
`234`	`234`
`235`	`235`	`ORT_RETURN_IF_NOT(is_nchw,`
`236`	`236`	`"Resize 'Cubic' mode only supports NCWH layout "`