Skip to content

Commit f059315

Browse files
minor
1 parent c1564f0 commit f059315

File tree

3 files changed

+4
-5
lines changed

3 files changed

+4
-5
lines changed

.ci/cspell_dict.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -223,9 +223,9 @@ logicalor
223223
logicalxor
224224
logit
225225
loglikelihoods
226+
lspec
226227
lstmsequence
227228
lstsq
228-
lspec
229229
lyalyushkin
230230
mapillary
231231
maskrcnn

nncf/onnx/graph/model_transformer.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,9 @@ class ONNXModelTransformer(ModelTransformer):
4747
ZERO_POINT_NAME_PREFIX = "zero_point_"
4848

4949
def __init__(self, model: onnx.ModelProto):
50-
super().__init__(model)
51-
self.onnx_model_extractor = onnx.utils.Extractor(model)
50+
inferred_model = onnx.shape_inference.infer_shapes(model)
51+
super().__init__(inferred_model)
52+
self.onnx_model_extractor = onnx.utils.Extractor(inferred_model)
5253

5354
def _get_target_edge(
5455
self,

nncf/onnx/quantization/quantize_model.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
import sys
1313
from pathlib import Path
1414
from typing import Any, Callable, Iterable, Optional, TypeVar, Union
15-
import tempfile
1615

1716
import onnx
1817
from onnx.external_data_helper import ExternalDataInfo
@@ -29,7 +28,6 @@
2928
from nncf.onnx.graph.model_metadata import MetadataKey
3029
from nncf.onnx.graph.model_metadata import remove_metadata
3130
from nncf.onnx.graph.model_metadata import set_metadata
32-
from nncf.onnx.graph.model_utils import eliminate_nop_cast
3331
from nncf.onnx.graph.nncf_graph_builder import GraphConverter
3432
from nncf.onnx.quantization.backend_parameters import get_external_data_dir
3533
from nncf.parameters import BackupMode

0 commit comments

Comments
 (0)