TODO to use find_nodes instead of get_graph_node_by_name

openvinotoolkit · Jul 26, 2024 · 12eb153 · 12eb153
1 parent 3b4a992
commit 12eb153
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nncf/experimental/torch/fx/node_utils.py b/nncf/experimental/torch/fx/node_utils.py
@@ -12,6 +12,8 @@
 import torch
 
 
+# TODO(dlyakhov): Use torch.fx.graph.find_nodes method instead after
+# torch version update (>= 2.4)
 def get_graph_node_by_name(graph: torch.fx.Graph, name: str) -> torch.fx.Node:
     """
     Retrieves a node with the specified name from the grpah.