2 Commits

Author	SHA1	Message	Date
Chengji Yao	a1cc9f33a3	[TPU] remove transpose ops in moe kernel (#18923) ... Signed-off-by: Chengji Yao <chengjiyao@google.com>	2025-05-29 23:00:11 +00:00
Michael Goin	e50a1f1a9c	[TPU] Add kernel test for moe_pallas (#17496) ... Signed-off-by: Michael Goin <mgoin64@gmail.com>	2025-05-06 17:59:57 -07:00