Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[ONNX] Update unsafe_chunk() method to support new version 13 of Spli…
…t operator. (#51415) (#51524) Summary: Pull Request resolved: #51524 * def unsafe_chunk() support and test in ops13. * Use _unsqueeze_helper insteadof Unsqueeze operator * Cast the splits into long. * Change the test to a fixed dimension. * Update test_pytorch_onnx_onnxruntime.py * Disable test_loop_with_list for opset 13. Test Plan: Imported from OSS Reviewed By: pbelevich Differential Revision: D26203123 Pulled By: SplitInfinity fbshipit-source-id: b273aeff8339faa0e8e9f1fcfbf877d1b703209f Co-authored-by: Negin Raoof <neginmr@utexas.edu>
- Loading branch information
1 parent
8ae6b0c
commit ba824eb
Showing
2 changed files
with
37 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters