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 6286f65 commit 9a44260Copy full SHA for 9a44260
src/libtorchaudio/rnnt/autograd.cpp
@@ -1,5 +1,4 @@
1
#include <libtorchaudio/rnnt/compute.h>
2
-#include <torch/script.h>
3
4
namespace torchaudio {
5
namespace rnnt {
src/libtorchaudio/rnnt/compute.cpp
std::tuple<torch::Tensor, std::optional<torch::Tensor>> rnnt_loss(
torch::Tensor& logits,
0 commit comments