Commit Graph

2 Commits

Author SHA1 Message Date
Justin Chu
5dd1c67776 [ONNX] Format ONNX python with black
Format all onnx python code with black and isort with

```sh
isort torch/onnx/ test/onnx
black torch/onnx/ test/onnx
```

Updated lintrunner config to include these paths.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76754
Approved by: https://github.com/suo, https://github.com/BowenBao
2022-05-05 00:19:22 +00:00
BowenBao
341e20a1b6 [ONNX] Add module name as PythonOp attribute (#67193) (#73281)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/73281

* Add module name as pythonOp attr

* Move to trace_post_record

* Add tests

* Code compactness

Test Plan: Imported from OSS

Reviewed By: jbschlosser

Differential Revision: D34625647

Pulled By: malfet

fbshipit-source-id: b04b2a4f1dc2cf733fcf50a3b022337f80d6eead
(cherry picked from commit 56e8658974e0a5f7faab211d51b3e425886bff8a)
2022-03-09 14:26:18 +00:00