@@ -64,7 +64,7 @@ def __init__(
64
64
inference_output_type = None ,
65
65
supported_ops = None ,
66
66
supported_types = None ,
67
- _experimental_new_quantizer = None ,
67
+ experimental_new_quantizer = None ,
68
68
):
69
69
"""Constructs QuantizationConfig.
70
70
@@ -87,7 +87,7 @@ def __init__(
87
87
Supported values are types exported by lite.constants. Frequently, an
88
88
optimization choice is driven by the most compact (i.e. smallest) type
89
89
in this list (default [constants.FLOAT]).
90
- _experimental_new_quantizer : Whether to enable experimental new quantizer.
90
+ experimental_new_quantizer : Whether to enable experimental new quantizer.
91
91
"""
92
92
93
93
if optimizations is None :
@@ -112,7 +112,7 @@ def __init__(
112
112
supported_types = [supported_types ]
113
113
self .supported_types = supported_types
114
114
115
- self ._experimental_new_quantizer = _experimental_new_quantizer
115
+ self .experimental_new_quantizer = experimental_new_quantizer
116
116
117
117
@classmethod
118
118
def create_dynamic_range_quantization (cls ,
@@ -198,6 +198,6 @@ def get_converter_with_quantization(self, converter, preprocess=None):
198
198
if self .supported_types :
199
199
converter .target_spec .supported_types = self .supported_types
200
200
201
- if self ._experimental_new_quantizer is not None :
202
- converter ._experimental_new_quantizer = self ._experimental_new_quantizer # pylint: disable=protected-access
201
+ if self .experimental_new_quantizer is not None :
202
+ converter .experimental_new_quantizer = self .experimental_new_quantizer
203
203
return converter
0 commit comments