Add ctc_greedy_decoder op (#5294)
* Add ctc_greedy_decoder op
* Update gpu kernel
* Update example
* Performance optimizing
* Add OF_CUDA_CHECK
* Code optimization
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/ops/loss_ops.py 73 additions, 1 deletiononeflow/python/ops/loss_ops.py
- oneflow/python/test/ops/test_ctc_greedy_decoder.py 166 additions, 0 deletionsoneflow/python/test/ops/test_ctc_greedy_decoder.py
- oneflow/user/kernels/ctc_greedy_decoder.cpp 57 additions, 0 deletionsoneflow/user/kernels/ctc_greedy_decoder.cpp
- oneflow/user/kernels/ctc_greedy_decoder.cu 144 additions, 0 deletionsoneflow/user/kernels/ctc_greedy_decoder.cu
- oneflow/user/kernels/ctc_greedy_decoder.h 72 additions, 0 deletionsoneflow/user/kernels/ctc_greedy_decoder.h
- oneflow/user/ops/ctc_loss_op.cpp 31 additions, 0 deletionsoneflow/user/ops/ctc_loss_op.cpp
Please register or sign in to comment