We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 003b3a9 commit 7727ad7Copy full SHA for 7727ad7
src/libtorchaudio/rnnt/autograd.cpp
src/libtorchaudio/rnnt/compute.cpp
@@ -30,4 +30,5 @@ TORCH_LIBRARY_FRAGMENT(torchaudio, m) {
30
"int blank,"
31
"float clamp,"
32
"bool fused_log_softmax) -> (Tensor, Tensor?)");
33
+ m.def("torchaudio::rnnt_loss_forward", &rnnt_loss);
34
}
0 commit comments