[Pallas] Support tgmm (#7137)

Commit

1 year ago

[Pallas] Support tgmm (#7137) Summary: This is to support Pallas' tgmm. Test Plan: python test/test_gmm.py