Commit Graph

6 Commits (c308c88d71e183e3ae79075c94f4ee5f72982fa8)

Author SHA1 Message Date
Yiqun Liu 3008fa1261
Add the CUDA kernel for beam_search op (#15020)
7 years ago
Tao Luo e0d4e04bdd fix some compiler warning
7 years ago
minqiyang 9725db0d40 Fix copy wrong pos bug
7 years ago
minqiyang 14ebc424d6 Add gpu support for unittest
7 years ago
minqiyang bd5a82e193 Polish unit test code
7 years ago
minqiyang 047fa2f9aa Add unit-test for sequence_pooling functor
7 years ago