fix avgpool export

Xilinx · Dec 31, 2024 · 18fee55 · 18fee55
1 parent 2ec8416
commit 18fee55
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/brevitas/proxy/runtime_quant.py b/src/brevitas/proxy/runtime_quant.py
@@ -292,6 +292,7 @@ def forward(self, x: IntQuantTensor) -> Union[Tensor, IntQuantTensor]:
             if self.export_mode:
                 out_tuple = self.export_handler(
                     x.value, x.scale, x.zero_point, x.bit_width, x.signed)
+                out_value, out_scale, out_zp, out_bit_width = out_tuple
             else:
                 out_tuple = self.tensor_quant(x.value, x.scale, x.zero_point, x.bit_width)
                 out_value, out_scale, out_zp, out_bit_width = out_tuple