[ONNX] Fix for constant folding flaky tests (#32546)
Summary:
Fix for constant folding flaky tests
Looks like the constant folding test modules are sometimes exported with ONNX_ATEN op export type, which is causing the CI failures.
I'm unable to repro this issue locally, but my guess is that the op export param is being overwritten on CI build at some point.
This PR sets the op export type and hopefully fixes the issue.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/32546
Reviewed By: hl475
Differential Revision: D19606919
Pulled By: houseroad
fbshipit-source-id: 31793d6857bbbf99b43b4a7c22a045a56ae19e44
