csukuangfj / optimized_transducer

Memory efficient transducer loss computation
Other
68 stars 12 forks source link

Implement RowSplitsToRowIds. #6

Closed csukuangfj closed 2 years ago

csukuangfj commented 2 years ago

The implementation is borrowed from

https://github.com/k2-fsa/k2/blob/master/k2/csrc/utils.cu#L75