[CUDA] Hotfix Scale with 1 parameter

2026-07-29 23:33:05 +04:00 · 2023-12-22 15:45:18 +03:00
parent a050285058
commit 938bc4d503
1 changed files with 3 additions and 0 deletions
@@ -128,6 +128,9 @@ namespace cv { namespace dnn { namespace cuda4dnn {

            /* the scale shift operation might require broadcasting */
            const int end_axis = [&] {
+                if (num_parameters == 1) {
+                    return static_cast<int>(axis + 1);
+                }
                for (int endAxis = axis + 1; endAxis <= input.rank(); endAxis++) {
                    if (input.size_range(axis, endAxis) == mid_size)
                        return endAxis;