xla

[Kernel] add group_offset and transpose_rhs support in gmm kernel

#9251

Merged