Zeeshan Siddiqui
6c0f447b51
Remove ONNX BatchNorm(12) test and converter. ( #37309 )
...
Summary:
Pursuant to https://github.com/onnx/onnx/pull/2750 we must remove PyTorch ONNX exporter related changes to BatchNorm(12) that were introduced as part of https://github.com/pytorch/pytorch/pull/35567 . This change is also needed to unblock ONNX [BUILD CI failures](https://circleci.com/gh/onnx/onnx/4629?utm_campaign=vcs-integration-link&utm_medium=referral&utm_source=github-build-link ) caused by PyTorch/Caffe2 tests.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37309
Reviewed By: hl475
Differential Revision: D21288914
Pulled By: houseroad
fbshipit-source-id: 15b076a2af55918dcd57f4e2fc77accd3d1510bd
2020-04-28 17:45:01 -07:00
Ksenija Stanojevic
92e91cee8d
ONNX Export Support for CrossEntropyLoss ( #34830 )
...
Summary:
Add ONNX export support for torch.nn.CrossEntropyLoss.
This PR makes following changes:
1. Updates nll_loss export
2. Makes a post pass for SoftmaxCrossEntropy
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34830
Reviewed By: hl475
Differential Revision: D21230712
Pulled By: houseroad
fbshipit-source-id: c81911a41968e23813ba10274340ce4d8ba1ed78
2020-04-25 17:56:53 -07:00
Negin Raoof
cb27067b32
[ONNX] Remove inverse op ( #37005 )
...
Summary:
ONNX inverse op is being removed.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37005
Reviewed By: hl475
Differential Revision: D21230728
Pulled By: houseroad
fbshipit-source-id: 7e10414918c57938cda4ca03875c070319d429fb
2020-04-25 12:23:15 -07:00
Negin Raoof
681ca45717
[ONNX] Export torch.inverse op ( #35318 )
...
Summary:
Added support for torch.inverse export as part of opset 12
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35318
Reviewed By: hl475
Differential Revision: D20865900
Pulled By: houseroad
fbshipit-source-id: be12b5194d21408cae24eec16e9e12377e8546ad
2020-04-07 10:48:33 -07:00
Lara Haidar
728c7dcea3
ONNX Update training ops and training amenable export API ( #35567 )
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/35567
Reviewed By: hl475
Differential Revision: D20715339
Pulled By: houseroad
fbshipit-source-id: ad88097e76b169035ab5814b769dc1bed54c6008
2020-03-29 23:14:25 -07:00
Alban Desmaison
45e1be9762
Revert D19710370: [pytorch][PR] ONNX Update training ops and training amenable export API
...
Test Plan: revert-hammer
Differential Revision:
D19710370
Original commit changeset: e5e79d385529
fbshipit-source-id: d0114dc561a3415869805d3fbf43b92730bbcf54
2020-03-27 06:51:05 -07:00
Lara Haidar
025a0abe5a
ONNX Update training ops and training amenable export API ( #32950 )
...
Summary:
- Update Dropout and Batchnorm in opset 12 : https://github.com/onnx/onnx/pull/2568
- Update api logic for exporting to ONNX training amenable models
Pull Request resolved: https://github.com/pytorch/pytorch/pull/32950
Reviewed By: hl475
Differential Revision: D19710370
Pulled By: houseroad
fbshipit-source-id: e5e79d38552936966662c41d39ddf33be1ba3e35
2020-03-27 00:39:39 -07:00
Lara Haidar
8216d9ae64
ONNX Export Support for NLLLoss ( #33509 )
...
Summary:
Adding ONNX export support for torch.nn.NLLLoss().
Pull Request resolved: https://github.com/pytorch/pytorch/pull/33509
Reviewed By: hl475
Differential Revision: D20052212
Pulled By: houseroad
fbshipit-source-id: 62efcff4efa1e0e97c65ad1b670c2fc1da08d28f
2020-03-05 11:13:21 -08:00
neginraoof
e03e4f3a2d
[ONNX] Add einsum export ( #32716 )
...
Summary:
Adding symbolic for onnx einsum as part of opset 12
Pull Request resolved: https://github.com/pytorch/pytorch/pull/32716
Reviewed By: hl475
Differential Revision: D19626168
Pulled By: houseroad
fbshipit-source-id: d8cc8af5f05f36aca3cd55dead602261ccdfec51
2020-02-03 12:56:50 -08:00
