Commit Graph

2 Commits (d54005a7f43af4107aa117fbd517f81c025165b3)

Author SHA1 Message Date
Yiqun Liu 16d54f7f23
Return parent_idx in beam_search op (#15520)
7 years ago
Yiqun Liu 3008fa1261
Add the CUDA kernel for beam_search op (#15020)
7 years ago