Add docs

openvinotoolkit · Aug 29, 2024 · ba76dbc · ba76dbc
1 parent 676983f
commit ba76dbc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/nncf/quantization/algorithms/min_max/backend.py b/nncf/quantization/algorithms/min_max/backend.py
@@ -36,7 +36,8 @@ class MinMaxAlgoBackend(ABC):
     @abstractmethod
     def preserved_metatypes(self) -> List[OperatorMetatype]:
         """
-        TODO
+        Property for backend-specific metatypes that require preserving float subgraphs
+        when removing the ShapeOf subgraph.
         """
 
     @property

diff --git a/nncf/quantization/passes.py b/nncf/quantization/passes.py
@@ -100,10 +100,11 @@ def find_preserved_nodes(
     preserved_metatypes: List[OperatorMetatype],
 ) -> List[NNCFNode]:
     """
-    :param graph:
-    :param shapeof_subgraphs:
-    :param preserved_metatypes:
-    :return:
+    :param graph: The input graph to be analyzed.
+    :param shapeof_subgraphs: A list of nodes belonging to ShapeOf subgraphs.
+    :param preserved_metatypes: Backend-specific metatypes that require preserving
+        float subgraphs when removing the ShapeOf subgraph.
+    :return: A list of nodes in float subgraphs of ShapeOf subgraphs.
     """
     preserved_nodes = set()
     for node in graph.get_nodes_by_metatypes(preserved_metatypes):