pytorch
daca0ee5 - [ONNX] Introduce ONNXScopeName (#82038)

Commit
2 years ago
[ONNX] Introduce ONNXScopeName (#82038) Update `_setup_trace_module_map` to always record module/layer info in `Scope` attribute for nodes. Extend `Scope` name to not only record module typename, but also module object variable name. Both names are formatted and stored as `name` attribute in `Scope`. Introduce `ONNXScopeName` class to manage the formatting and parsing. Updated local function export code adjusting to this update. Pull Request resolved: https://github.com/pytorch/pytorch/pull/82038 Approved by: https://github.com/AllenTiTaiWang, https://github.com/justinchuby, https://github.com/abock, https://github.com/malfet
Author
Committer
Parents
Loading