[PT FE] Improve operation naming (#17997)

* Improve operation naming

* Use set to reduce operations with indexes, fix code style

* Refactor

* Make good names in transformations

* Remove tensor index from input/output names

* Fix tests
This commit is contained in:
Maxim Vafin authored and GitHub committed 2023-06-15 14:28:38 +02:00
1 parent 483a040d52
commit 9684f9184a
21 files changed
+354 -314

No files matched your search

@@ -319,6 +319,13 @@ class TorchScriptPythonDecoder (Decoder):
return self.outputs()[index]
def mark_node(self, node):
name = self.graph_element.kind()
if "FrameworkNode" not in node.get_type_name():
name += "/" + node.get_type_name()
if self.graph_element.scopeName():
node.set_friendly_name(self.graph_element.scopeName().split("/")[-1] + "/" + name)
else:
node.set_friendly_name(name)
return node
def try_decode_get_attr(self):