[OpenVINOQuantizer] Mark quantizer and quantize_pt2e as API

Author: daniil-lyakhov

nncf/experimental/torch/fx/__init__.py

@@ -8,3 +8,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from nncf.experimental.torch.fx.quantization.quantize_pt2e import quantize_pt2e as quantize_pt2e
+from nncf.experimental.torch.fx.quantization.quantizer.openvino_quantizer import OpenVINOQuantizer as OpenVINOQuantizer

nncf/experimental/torch/fx/quantization/quantize_pt2e.py

@@ -25,6 +25,7 @@
 import nncf
 from nncf.common.factory import NNCFGraphFactory
 from nncf.common.logging import nncf_logger
+from nncf.common.utils.api_marker import api
 from nncf.data import Dataset
 from nncf.experimental.quantization.algorithms.post_training.algorithm import ExperimentalPostTrainingQuantization
 from nncf.experimental.torch.fx.constant_folding import constant_fold
@@ -38,6 +39,7 @@
 from nncf.quantization.advanced_parameters import RangeEstimatorParameters
 
 
+@api()
 def quantize_pt2e(
     model: torch.fx.GraphModule,
     quantizer: Quantizer,

nncf/experimental/torch/fx/quantization/quantizer/openvino_quantizer.py

@@ -30,6 +30,7 @@
 from nncf.common.quantization.quantizer_setup import SingleConfigQuantizerSetup
 from nncf.common.quantization.structs import QuantizationPreset
 from nncf.common.quantization.structs import QuantizationScheme
+from nncf.common.utils.api_marker import api
 from nncf.experimental.torch.fx.nncf_graph_builder import GraphConverter
 from nncf.experimental.torch.fx.node_utils import get_graph_node_by_name
 from nncf.experimental.torch.fx.transformations import fold_constant_except_qdq
@@ -46,6 +47,7 @@
 QUANT_ANNOTATION_KEY = "quantization_annotation"
 
 
+@api()
 class OpenVINOQuantizer(TorchAOQuantizer):
     """
     Implementation of the Torch AO quantizer which annotates models with quantization annotations