ContextQMD
Back to Tensorrt

**tensorflow_quantization.CustomQDQInsertionCase**

tools/tensorflow-quantization/docs/source/cqdq.rst

23.08842 B
Original Source

.. _cqdq_api:

tensorflow_quantization.CustomQDQInsertionCase

.. autoclass:: tensorflow_quantization.CustomQDQInsertionCase :members:

Example

.. code:: python

class EfficientNetQDQCase(CustomQDQInsertionCase): def init(self) -> None: super().init()

def info(self):
    return "In Multiply operation quantize inputs at index 0 and 1."

def case(self, keras_model: 'tf.keras.Model', qspec: 'QuantizationSpec') -> 'QuantizationSpec':
    se_block_qspec_object = QuantizationSpec()
    for layer in keras_model.layers:
        if isinstance(layer, tf.keras.layers.Multiply):
            se_block_qspec_object.add(layer.name, quantize_input=True, quantize_weight=False, quantization_index=[0, 1])
    return se_block_qspec_object