Add interface supporting double type

baidu-research / warp-ctc

Fast parallel CTC.

Apache License 2.0

4.06k stars 1.04k forks source link

Add interface supporting double type #163

Closed lfchener closed 4 years ago

lfchener commented 4 years ago

1、add interface compute_ctc_loss_double to support double probabilities. 2、add interface get_workspace_size_double to support get workspace when probability is double. 3、add template for reduce to support compute_ctc_loss_double on gpu.