Commit Graph

5 Commits (16ec4b8c8bd5c95c38c39d2a2528027b4a1930b6)

Author SHA1 Message Date
Yiqun Liu 3008fa1261
Add the CUDA kernel for beam_search op (#15020)
7 years ago
frankwhzhang c9a653820b fix label_pos ,add test_layers.py, test=develop
7 years ago
frankwhzhang a672b291e5 fix code style, test=develop
7 years ago
frankwhzhang 97de98cd0a update bpr_loss op code, test=develop
7 years ago
frankwhzhang 570d89ec84 add bpr_loss operator , test=develop
7 years ago