From 7bc0e3d7ebddefdb98b838e34f782e805578c551 Mon Sep 17 00:00:00 2001 From: Alexander Suslov Date: Thu, 3 Aug 2023 13:13:58 +0400 Subject: [PATCH] replied to comments --- nncf/onnx/graph/model_utils.py | 1 + nncf/openvino/graph/model_utils.py | 1 + 2 files changed, 2 insertions(+) diff --git a/nncf/onnx/graph/model_utils.py b/nncf/onnx/graph/model_utils.py index 62facd50614..41c9afcfe54 100644 --- a/nncf/onnx/graph/model_utils.py +++ b/nncf/onnx/graph/model_utils.py @@ -28,6 +28,7 @@ def remove_fq_from_inputs(model: onnx.ModelProto, nncf_graph: NNCFGraph) -> onnx It's needed for the further bias shift calculation that relates on quantized weights. :param model: onnx.ModelProto instance. + :param nncf_graph: NNCFGraph instance. :return: onnx.ModelProto instance without activation Quantizer nodes. """ transformation_layout = TransformationLayout() diff --git a/nncf/openvino/graph/model_utils.py b/nncf/openvino/graph/model_utils.py index aac584c321b..803b72aa811 100644 --- a/nncf/openvino/graph/model_utils.py +++ b/nncf/openvino/graph/model_utils.py @@ -56,6 +56,7 @@ def remove_fq_from_inputs(model: ov.Model, graph: NNCFGraph) -> ov.Model: It's needed for the further bias shift calculation that relates on quantized weights. :param model: ov.Model instance. + :param graph: NNCFGraph instance. :return: ov.Model instance without activation Fake Quantize nodes. """ transformation_layout = TransformationLayout()