Fix rebase

daniil-lyakhov · Nov 14, 2023 · 472007d · 472007d
1 parent 7744955
commit 472007d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nncf/quantization/algorithms/smooth_quant/algorithm.py b/nncf/quantization/algorithms/smooth_quant/algorithm.py
@@ -360,7 +360,9 @@ def _process_weight_statistics(self, node: NNCFNode, weights: TTensor, port_id:
         channel_axis = 0
         if len(weights.shape) > 1:
             channel_axis = self._backend_entity.get_weight_channel_axis(node)
-        return self._backend_entity.process_weight_statistics(weights, channel_axis)
+        reduction_shape = [i for i, _ in enumerate(weights.shape)]
+        reduction_shape.pop(channel_axis)
+        return self._backend_entity.process_weight_statistics(weights, tuple(reduction_shape))
 
     def _create_scale_node_name(self, source_name: str, source_port_id: int) -> str:
         """