Make JIT Aliasing Test Less Brittle (#65493)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/65493
Added a last resolve to use whatever ATen operator that has Tensor outputs in the graph as the operator node to check alias annotation.
Test Plan:
python test/test_ops.py -k test_variant_consistency_jit_linalg_tensorinv
python test/test_ops.py -k test_variant_consistency_jit_nn_functional_normalize
Reviewed By: eellison
Differential Revision: D31132861
Pulled By: alanwaketan
fbshipit-source-id: 26fc2e6bc77be3a296967cf29a3f6ded231302fa