fix: Fixed perCh path in sawb_params_code

Signed-off-by: Brandon Groth <[email protected]>

Author: BrandonGroth

fms_mo/quant_refactor/sawb_utils.py

@@ -110,10 +110,10 @@ def sawb_params_code(
 
         if perCh:
             # per-channel
-            reduce_dim = list(range(1, len(input.shape)))
+            reduce_dim = list(range(1, len(input_tensor.shape)))
             # conv W=[ch_o, ch_i, ki, ij], linear W=[ch_o, ch_i], reduce all dim but ch_out
-            mu = torch.mean(input.abs(), dim=reduce_dim)
-            std = torch.mean(input**2, dim=reduce_dim).sqrt()
+            mu = torch.mean(input_tensor.abs(), dim=reduce_dim)
+            std = torch.mean(input_tensor**2, dim=reduce_dim).sqrt()
             clip_val_vec = torch.tensor(coeff[1] * mu + coeff[0] * std)
             return None, clip_val_vec