Commit Graph

6 Commits (c34b24ede782612464bc4c7cad47c40661616e9d)

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