Fix (nn): disable weight tensor slicing syntax

Xilinx · Jun 20, 2023 · b579453 · b579453
1 parent 808bddc
commit b579453
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/nn/mixin/parameter.py b/src/brevitas/nn/mixin/parameter.py
@@ -103,7 +103,7 @@ def quant_weight(
             # The modules to slice should have been cached already at this point
             assert self._cached_sub_tensor_slice_list_modules is not None, "Missing cache of modules to slice."
             for m in self._cached_sub_tensor_slice_list_modules:
-                m.subtensor_slice_list = None
+                m.subtensor_slice_list = [None]
         return out
 
     def int_weight(self, float_datatype=False):