pytorch/torch/onnx
Sergii Dymchenko 5df8547ff9 Fix ONNX LogSoftmax export. (#9576)
Summary:
This fixes an issue with incorrect `axis=-1` in the exported ONNX.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9576

Reviewed By: yinghai

Differential Revision: D9125463

Pulled By: houseroad

fbshipit-source-id: 6f4cb1067d1aa6bb0a9f56690fc21816c98eebfa
2018-08-03 22:09:42 -07:00
..
__init__.py Make JIT tracing a thread-local property (#9414) 2018-07-19 19:09:39 -07:00
operators.py Fixes for interpreter and ONNX export for translation (#7044) 2018-04-27 22:23:57 -07:00
symbolic.py Fix ONNX LogSoftmax export. (#9576) 2018-08-03 22:09:42 -07:00
utils.py Add (constant size) TensorLists to JIT, use them in cat and stack nodes (#9948) 2018-07-31 07:39:52 -07:00