Restore Conv1d Channel Last Fix

Summary: D93658558 accidentally dropped the conv1d check which was added shortly before it, affecting indexing and resulting in incorrect results

Differential Revision: D96488193

Author: JakeStevens

backends/cadence/generic/operators/op_quantized_conv2d.cpp

@@ -511,13 +511,13 @@ void quantized_conv2d_nhwc(
   // Depthwise is defined by in_channels == groups; depthwise weights have one
   // fewer dim than regular weights because the IC dim (always 1) was squeezed.
   const bool is_depthwise =
-      !conv1d && c == groups && weight.dim() < input.dim();
+      c == groups && weight.dim() < input.dim();
   int oc, wh, ww, wc;
   if (is_depthwise) {
-    // Depthwise weight is [KH, KW, OC]
-    wh = static_cast<int>(weight.size(0));
-    ww = static_cast<int>(weight.size(1));
-    oc = static_cast<int>(weight.size(2));
+    // Depthwise weight: conv2d=[KH, KW, OC], conv1d=[K, OC]
+    wh = static_cast<int>(conv1d ? 1 : weight.size(0));
+    ww = static_cast<int>(conv1d ? weight.size(0) : weight.size(1));
+    oc = static_cast<int>(conv1d ? weight.size(1) : weight.size(2));
     wc = 1;
   } else {
     // Regular weight is [OC, WH, WW, WC] or for conv1d [OC, WW, WC]

backends/cadence/hifi/operators/op_quantized_conv2d_nhwc_out.cpp

@@ -184,16 +184,16 @@ void xa_opt_quantized_conv2d_nhwc(
     // fewer dim than regular weights because the IC dim (always 1) was
     // squeezed.
     bool is_depthwise =
-        !conv1d && input_channels == groups && weight.dim() < input.dim();
+        input_channels == groups && weight.dim() < input.dim();
     WORD32 kernel_height;
     WORD32 kernel_width;
     WORD32 kernel_channels;
     WORD32 out_channels;
     if (is_depthwise) {
-      // Depthwise weight is [KH, KW, OC]
-      kernel_height = weight.size(0);
-      kernel_width = weight.size(1);
-      out_channels = weight.size(2);
+      // Depthwise weight: conv2d=[KH, KW, OC], conv1d=[K, OC]
+      kernel_height = conv1d ? 1 : weight.size(0);
+      kernel_width = conv1d ? weight.size(0) : weight.size(1);
+      out_channels = conv1d ? weight.size(1) : weight.size(2);
       kernel_channels = 1;
     } else {
       // Regular weight is [OC, IC, KH, KW] or for conv1d [OC, K, IC]
@@ -385,13 +385,13 @@ void quantized_conv2d_nhwc(
   // Depthwise is defined by in_channels == groups; depthwise weights have one
   // fewer dim than regular weights because the IC dim (always 1) was squeezed.
   const bool is_depthwise =
-      !conv1d && c == groups && weight.dim() < input.dim();
+      c == groups && weight.dim() < input.dim();
   int oc, wh, ww, wc;
   if (is_depthwise) {
-    // Depthwise weight is [KH, KW, OC]
-    wh = weight.size(0);
-    ww = weight.size(1);
-    oc = weight.size(2);
+    // Depthwise weight: conv2d=[KH, KW, OC], conv1d=[K, OC]
+    wh = conv1d ? 1 : weight.size(0);
+    ww = conv1d ? weight.size(0) : weight.size(1);
+    oc = conv1d ? weight.size(1) : weight.size(2);
     wc = 1;
   } else {
     // Regular weight is [OC, WH, WW, WC] or for conv1d [OC, WW, WC]