pytorch
75995e4b - [ONNX] Add support for hann_window operator. (#54587) (#56163)

Commit
4 years ago
[ONNX] Add support for hann_window operator. (#54587) (#56163) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56163 * [ONNX] Improve index_put symbolic to handle singular Bool updates (#53690) Adds support for cases where the updates to the index_put node is a single Bool value, such as the case shown below ``` mask[indices] = True ``` Fixes #53507 * [ONNX] Support primitive type input/outputs and attributes (#53550) Support primitive type attributes. Needed for Silero model. * [ONNX] Fix if output shape mismatch error & Fix graph input directly used as output (#53219) Fix if output shape mismatch error & Fix graph input directly used as output * Add support for hann_window operator. * [ONNX] Replace decomposeLinear pre process pass with a symbolic (#53077) Replace decomposeLinear pre process pass with a symbolic * Add a test case for dtype is None. * Resolve flake8 issue. * Remove one unused test case. * Add support for hann_window operator. * Add a test case for dtype is None. * Remove one unused test case. Test Plan: Imported from OSS Reviewed By: pbelevich Differential Revision: D27866145 Pulled By: SplitInfinity fbshipit-source-id: e0b43df9ecd1a95cd7ac297213aba453bbaf2913 Co-authored-by: Shubham Bhokare <32080845+shubhambhokare1@users.noreply.github.com> Co-authored-by: Negin Raoof <neginmr@utexas.edu> Co-authored-by: Bowen Bao <bowbao@microsoft.com> Co-authored-by: Ksenija Stanojevic <KsenijaS@users.noreply.github.com>
Author
Parents
Loading