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 2e2421d commit c1e3840Copy full SHA for c1e3840
src/torchaudio/transforms/_transforms.py
@@ -10,7 +10,7 @@
10
from torch.nn.parameter import UninitializedParameter
11
12
from torchaudio import functional as F
13
-from torchaudio.functional.functional import _rnnt_loss
+from torchaudio.functional.functional import rnnt_loss
14
from torchaudio.functional.functional import (
15
_apply_sinc_resample_kernel,
16
_check_convolve_mode,
@@ -1847,7 +1847,7 @@ def forward(
1847
Tensor: Loss with the reduction option applied. If ``reduction`` is ``"none"``, then size (batch),
1848
otherwise scalar.
1849
"""
1850
- return _rnnt_loss(
+ return rnnt_loss(
1851
logits,
1852
targets,
1853
logit_lengths,
0 commit comments