Removed unused check

Xilinx · Mar 19, 2024 · c6f5216 · c6f5216
1 parent 368d0a3
commit c6f5216
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/brevitas/nn/mixin/base.py b/src/brevitas/nn/mixin/base.py
@@ -75,7 +75,8 @@ def unpack_input(self, inp: Union[Tensor, QuantTensor]) -> Union[Tensor, QuantTe
         # when used as input to tracing (e.g. during ONNX export)
         if (torch._C._get_tracing_state() is not None and isinstance(inp, tuple) and
                 len(inp) == len(QuantTensor._fields) and all([isinstance(t, Tensor) for t in inp])):
-            inp = QuantTensor(*inp)
+            raise
+            # inp = QuantTensor(*inp)
         if not torch._C._get_tracing_state():
             if isinstance(inp, QuantTensor):
                 inp = inp.set(value=inp.value.rename(None))