fix for tensor

Xilinx · Dec 18, 2024 · d7b5036 · d7b5036
1 parent 8775ca2
commit d7b5036
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/proxy/runtime_quant.py b/src/brevitas/proxy/runtime_quant.py
@@ -227,7 +227,7 @@ def create_quant_tensor(
             qt_args: Union[Tensor, Tuple[Any]],
             x: Optional[IntQuantTensor] = None) -> IntQuantTensor:
 
-        if x is None:
+        if isinstance(qt_args, tuple):
             out = IntQuantTensor(*qt_args, self.is_signed, self.training)
         else:
             out = IntQuantTensor(