[XLA:GPU] Generalize propagation of dimensions in Triton GEMM rewriter.

Change the function interfaces to support HLOs with different dimension orders on operands. At this point this is ~NFC because the actual analysis for such HLOs will be added in next CLs.

PiperOrigin-RevId: 555928758

Author: Ilia Sergachev