fixes

Xilinx · Mar 4, 2024 · 969886e · 969886e
1 parent 56c5c7e
commit 969886e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/brevitas/export/onnx/standard/qoperator/handler/act.py b/src/brevitas/export/onnx/standard/qoperator/handler/act.py
@@ -20,9 +20,9 @@ class StdQOpONNXQuantNLALHandler(StdQOpONNXQuantLayerHandler, ABC):
 
     @classmethod
     def validate(cls, module: QuantNLAL):
-        if cls.input_quant_supported and module.is_input_quant_enabled:
+        if cls.input_quant_supported and module.input_quant.is_quant_enabled:
             assert not module.input_quant.is_quant_enabled, "Narrow range quant not supported."
-        elif not cls.input_quant_supported and module.is_input_quant_enabled:
+        elif not cls.input_quant_supported and module.input_quant.is_quant_enabled:
             raise RuntimeError("Input quant not supported.")
         if module.act_quant.is_quant_enabled:
             assert not module.act_quant.is_narrow_range, "Narrow range quant not supported."