Skip to content

Commit 58d5112

Browse files
committed
[NNCF] Missing import statements in utils.py
Changes: 1. Added missing import statements in nncf/common/graph/utils.py Reason for changes: (openvinotoolkit#3249) Torch and Tensorflow backend-specific methods need to be removed from common layer attributes and all related calls need to be replaced by their corresponding legacy function calls
1 parent eb7905f commit 58d5112

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

nncf/common/graph/utils.py

+5
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,11 @@
1414

1515
from nncf.common.graph import NNCFGraph
1616
from nncf.common.graph import NNCFNode
17+
from nncf.common.graph.layer_attributes import WeightedLayerAttributes
18+
from nncf.common.graph.layer_attributes import GenericWeightedLayerAttributes
19+
from nncf.common.graph.layer_attributes import LinearLayerAttributes
20+
from nncf.common.graph.layer_attributes import ConvolutionLayerAttributes
21+
from nncf.common.graph.layer_attributes import GroupNormLayerAttributes
1722
from nncf.common.graph.operator_metatypes import OperatorMetatype
1823
from nncf.common.logging import nncf_logger
1924
from nncf.common.pruning.utils import traverse_function

0 commit comments

Comments
 (0)