Keras int input error fix (#2815)

* Fixed error facing, when input to Keras model is Int type Signed-off-by: Rishabh Kumar Jain <quic_rij@quicinc.com>
quic · Mar 8, 2024 · e068c40 · e068c40
1 parent ed6645e
commit e068c40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...xtensions/tensorflow/src/python/aimet_tensorflow/keras/quantsim_config/quantsim_config.py b/...xtensions/tensorflow/src/python/aimet_tensorflow/keras/quantsim_config/quantsim_config.py
@@ -185,7 +185,7 @@ def _initialize_input_quantizers(layer: layers.Layer, quant_settings: QuantizerS
                                                                 quant_settings.is_symmetric,
                                                                 quant_settings.use_strict_symmetric,
                                                                 quant_settings.use_unsigned_symmetric,
-                                                                enabled and layer.output.dtype in QUANT_ALLOWED_DTYPES)
+                                                                enabled and (layer.input[i].dtype if num_inputs > 1 else layer.input.dtype) in QUANT_ALLOWED_DTYPES)
         input_quantizers.append(activation_tensor_quantizer)
     return input_quantizers