[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:
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):
|
||||
|
||||
Reference in new issue
Block a user