# /*---------------------------------------------------------------------------------------------
#  * Copyright (c) 2025 STMicroelectronics.
#  * All rights reserved.
#  *
#  * This software is licensed under terms that can be found in the LICENSE file in
#  * the root directory of this software component.
#  * If no LICENSE file comes with this software, it is provided AS-IS.
#  *--------------------------------------------------------------------------------------------*/
from common.registries.quantizer_registry import QUANTIZER_WRAPPER_REGISTRY

from image_classification.tf.src.quantization.tf_quantizer import TFLitePTQQuantizer

__all__ = ['TFLitePTQQuantizer']

# Register the TF PTQ Quantizer class from another folder
QUANTIZER_WRAPPER_REGISTRY.register(
    quantizer_name="tflite_converter",
    framework="torch",
    use_case="image_classification"
)(TFLitePTQQuantizer)