Skip to content

Commit 7727ad7

Browse files
committed
Move rnnt_loss_forward to compute.cpp
1 parent 003b3a9 commit 7727ad7

File tree

2 files changed

+1
-10
lines changed

2 files changed

+1
-10
lines changed

src/libtorchaudio/rnnt/autograd.cpp

Lines changed: 0 additions & 10 deletions
This file was deleted.

src/libtorchaudio/rnnt/compute.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,4 +30,5 @@ TORCH_LIBRARY_FRAGMENT(torchaudio, m) {
3030
"int blank,"
3131
"float clamp,"
3232
"bool fused_log_softmax) -> (Tensor, Tensor?)");
33+
m.def("torchaudio::rnnt_loss_forward", &rnnt_loss);
3334
}

0 commit comments

Comments
 (0)