53 Commits

Author	SHA1	Message	Date
Justin Chu	def778527e	[ONNX] Quantization support for five ops (#78103) ... - Add quantization support for `interpolate`, `avgpool`, `sigmoid` and `add_relu` - Return the inputs to ListUnpack if the previous node is ListConstruct so that `ListConstruct` and `ListUnpack` are canceled and removed in the jit passes. ONNX doesn't support them. Pull Request resolved: https://github.com/pytorch/pytorch/pull/78103 Approved by: https://github.com/garymm	2022-06-03 20:22:07 +00:00
BowenBao	6883b0ce9f	[ONNX][WIP] Refactor verification.py ... * Rename arguments, code clean up. * Refactor functions to smaller reusable functions. Pull Request resolved: https://github.com/pytorch/pytorch/pull/77289 Approved by: https://github.com/justinchuby, https://github.com/garymm	2022-05-31 18:49:39 +00:00
BowenBao	483bb4f0cb	[ONNX] Extract export verification as standalone api from unittest ... The verification logic is refactored and extracted from `test_pytorch_onnx_onnxruntime.py`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/76929 Approved by: https://github.com/justinchuby, https://github.com/garymm	2022-05-26 19:49:19 +00:00