pytorch/torch/onnx
Yael Dekel 3fa0b1e325 ONNX: fix bug in export of cumsum operator (#40044)
Summary:
The "cast" operator is currently added after the cumsum operator, but it should be added before, since torch.cumsum supports more types than ONNX (specifically, bool).
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40044

Reviewed By: hl475

Differential Revision: D22158013

Pulled By: houseroad

fbshipit-source-id: e6c706572b9b8de880d4d71eaa132744ef01ad4d
2020-06-22 10:11:35 -07:00
..
__init__.py [ONNX] Remove Aten ops from ONNX export (#37239) 2020-05-29 21:20:14 -07:00
operators.py Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
symbolic_caffe2.py [quant][onnx] Support conversion of quantized sigmoid operator from pytorch to caffe2 (#34629) 2020-03-13 22:42:06 -07:00
symbolic_helper.py ONNX: fix bug in export of ops involving torch.bool type (#40006) 2020-06-22 09:57:25 -07:00
symbolic_opset7.py Export Meshgrid (#26037) 2019-10-25 16:59:22 -07:00
symbolic_opset8.py Renaming scales parameter for interpolate (#31526) 2020-01-02 08:19:30 -08:00
symbolic_opset9.py Fix ONNX export of RNNs with no bias (#36894) 2020-06-08 20:36:22 -07:00
symbolic_opset10.py [ONNX] Fix export for avg_pool with default stride (#33017) 2020-02-07 22:46:46 -08:00
symbolic_opset11.py ONNX: fix bug in export of cumsum operator (#40044) 2020-06-22 10:11:35 -07:00
symbolic_opset12.py [ONNX] Update black_listed_operators for opset 12 (#39414) 2020-06-19 13:33:25 -07:00
symbolic_registry.py ONNX Error Message on Missing Op (#33593) 2020-02-26 15:18:16 -08:00
utils.py [ONNX] Fix regression disabling checker (#39073) 2020-06-11 14:03:18 -07:00