Add CTC Loss (#4034)
* Add CTC Loss
* Add backward kernel
* Remove tf in test
* Update api document
* Add zero_infinity option
* refine
* Add 1n2d test case
* Switch to consistent_view
* Fix Eager mode
* Remove duplicate license
* Add grad check
* Fix bw test
* Fix bugs
* Add op name
* Refine
* of_format
* Expand annotation
* Performance optimizing for cuda
* Check input_length & target_lengths
* Update __syncthreads
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/ops/loss_ops.py 123 additions, 0 deletionsoneflow/python/ops/loss_ops.py
- oneflow/python/test/ops/test_ctc_loss.py 352 additions, 0 deletionsoneflow/python/test/ops/test_ctc_loss.py
- oneflow/user/kernels/ctc_loss_kernel.cpp 142 additions, 0 deletionsoneflow/user/kernels/ctc_loss_kernel.cpp
- oneflow/user/kernels/ctc_loss_kernel_util.cpp 232 additions, 0 deletionsoneflow/user/kernels/ctc_loss_kernel_util.cpp
- oneflow/user/kernels/ctc_loss_kernel_util.cu 267 additions, 0 deletionsoneflow/user/kernels/ctc_loss_kernel_util.cu
- oneflow/user/kernels/ctc_loss_kernel_util.h 47 additions, 0 deletionsoneflow/user/kernels/ctc_loss_kernel_util.h
- oneflow/user/ops/ctc_loss_op.cpp 134 additions, 0 deletionsoneflow/user/ops/ctc_loss_op.cpp
Please register or sign in to comment