Fix for keepdim

Xilinx · Jun 19, 2024 · bcca84e · bcca84e
1 parent 500627d
commit bcca84e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/brevitas/quant/solver/common.py b/src/brevitas/quant/solver/common.py
@@ -184,6 +184,8 @@ def stats_reduce_dim(scaling_stats_op, scaling_per_output_channel, group_dim=Non
     def keepdim(group_dim=None):
         if group_dim is not None:
             return True
+        else:
+            return False
 
 
 class SolveScalingStatsInputViewShapeImplFromEnum(ExtendedInjector):