[ONNX] Add support for shrink ops (#66969) (#68492)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/68492
* Initial commit
* Fix flake issue
* Add test tags
Test Plan: Imported from OSS
Reviewed By: jansel
Differential Revision: D32483827
Pulled By: msaroufim
fbshipit-source-id: 41c623712524465b877d0fe0e2f4001d475bf2ce