File size: 921 Bytes
747451d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | # /*---------------------------------------------------------------------------------------------
# * 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)
|