OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
f77783c374
pytorch/torch/onnx
History
shubhambhokare1 671c8a459a [ONNX] Add pixel_unshuffle support in opset 9 
Current we are unable to utilize ONNX's SpaceToDepth operator due to the lack of the mode_s attribute, hence we add an alternative symbolic in opset 9 to support pixel_unshuffle

- Adds support for pixel_unshuffle in opset9
- Adds support for dynamic input shapes for pixel_shuffle and pixel_unshuffle
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72449
2022-02-19 00:15:16 +00:00
..
__init__.py [ONNX] Refactor _run_symbolic_function (#67573) (#68491) 2022-02-11 18:35:35 +00:00
operators.py Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
README.md [ONNX] Minor doc update (#69501) (#69550) 2022-02-11 22:05:15 +00:00
symbolic_caffe2.py [ONNX] use consistent quoting for string literals (#57757) (#58695) 2021-05-27 12:06:42 -07:00
symbolic_helper.py [ONNX] First version of quantized model export: Support quantized.Linear (#69232) 2022-02-18 18:27:26 +00:00
symbolic_opset7.py [ONNX] Fix remainder export (#64230) (#64578) 2021-09-30 21:08:54 -07:00
symbolic_opset8.py Added antialias flag to interpolate (CPU only, bilinear) (#65142) 2021-11-17 09:10:15 -08:00
symbolic_opset9.py [ONNX] Add pixel_unshuffle support in opset 9 2022-02-19 00:15:16 +00:00
symbolic_opset10.py [ONNX] First version of quantized model export: Support quantized.Linear (#69232) 2022-02-18 18:27:26 +00:00
symbolic_opset11.py [ONNX] Add export support for linalg norm (#66575) 2022-02-18 18:30:16 +00:00
symbolic_opset12.py [ONNX] Support opset 15 (#67121) (#67805) 2021-11-08 14:37:00 -08:00
symbolic_opset13.py [ONNX] Use human readable enum for dtype scalars (#66822) (#67807) 2021-11-08 14:37:05 -08:00
symbolic_opset14.py [ONNX] Fix for batchnorm training op mode (#52758) (#62760) 2021-08-18 13:29:07 -07:00
symbolic_opset15.py [ONNX] Support opset 15 (#67121) (#67805) 2021-11-08 14:37:00 -08:00
symbolic_registry.py [ONNX] First version of quantized model export: Support quantized.Linear (#69232) 2022-02-18 18:27:26 +00:00
utils.py Integrate full ONNX check into ONNX export API (#71125) 2022-02-18 18:40:09 +00:00

README.md

torch.onnx

Torch->ONNX converter / exporter.

User-facing docs.

Developer docs.