[xla:gpu] Fix the dimension order propagation for kPad
[xla:gpu] Fix the dimension order propagation for kPad

Previously trivial fragments are not handled because we didn't support matrices with trivial dimensions.
