tree 2449875cd087e63a784600c73919abcc9f093d27
parent 3592c594ae611115987b943019ebbbaf61efce9b
author NatashaKnk <natashaknk@google.com> 1691171731 -0700
committer GitHub <noreply@github.com> 1691171731 -0700
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsBcBAABCAAQBQJkzTuTCRBK7hj4Ov3rIwAAVRsIAFmvzNIcxjs70tnq6kXlznOa
 fw1dmIeNrLWRCI1jXEjNu9SrXzh1tRFYrUVmpiijYpoOyf9fW+iGBdHnjkbEiC5I
 cZ3kC3+BKqdBAGWolfn78TN1DS8dWFw0oErChRQB7NnE5r2KopyKopGdfhPfrcRu
 p94GMoNzHCJ1x06g+iP8lD5dOP4PX83EJlZ0P5bpFaOctF3u1wYyPE4T7DdLrsGd
 ktWXSiDuqf8DONL3unfRTKqkR3mfzabDrMKaTWF1fDLoLay8TTaM1fFrLATeA1HU
 mKSAS9HkyEFRfUnFvEeUAoOFerGGn64bk+MCpw8S7w0L9qV+LzMOSlilovowfsw=
 =eTqS
 -----END PGP SIGNATURE-----
 

Add canonicalizer that reorders unary elementwise ops and shape manipulation ops. (#14494)

The pass identifies shape manipulation operations (`reshape`,
`transpose`, `broadcast`) that feed into unary elementwise operations
and swaps their order. Since elementwise ops do not manipulate the input
shape (and there is no interaction between the elements), the reordering
doesn't affect the result.
This enables a future PR where some consecutive shape manipulation
operations should be collapsed together into a single operation.
`Broadcast_in_dim` is not included in this PR as the same thing is done
for it as part of an existing `StablehloToStablehlo` pass.
In the future, we might want to consider merging/more clearly defining
the difference between `Canonicalizations.cpp` and
`StablehloToStablehlo.cpp` files, as at the moment their function is
almost indistinguishable in some cases.