Resubmit PR-18512: Improved onnx export for 3 onnx ops (#18571)
Summary:
Fix ROCm CI failure
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18571
Differential Revision: D14669323
Pulled By: bddppq
fbshipit-source-id: 022afe5c20e680295c9cfdfe1ec14650305955a8