PR #11551: Dimensional Ordering of Layer Norm Operands
PR #11551: Dimensional Ordering of Layer Norm Operands

Imported from GitHub PR https://github.com/openxla/xla/pull/11551

Orders the dimensions of the operands of layer norm Custom Calls as required by cuDNN 9.1 and newer.
Copybara import of the project:

--
d9369e905968697139b6c3d7ecf12698abcc1e04 by Philipp Hack <phack@nvidia.com>:

Aligns the ordering of the dimensions of layer norm operands.

Merging this change closes #11551

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/11551 from philipphack:u_layer_dimensions_xla d9369e905968697139b6c3d7ecf12698abcc1e04
