2 Commits

Author	SHA1	Message	Date
Ying Sheng	c98e84c21e	[Minor, Performance] Use torch.argmax for greedy sampling (#1589)	2024-10-06 13:15:05 -07:00
Ke Bao	33b54e7c40	Add pytorch sampling backend ut (#1425)	2024-09-15 01:15:30 +10:00