diff --git a/src/libtorchaudio/rnnt/cpu/compute.cpp b/src/libtorchaudio/rnnt/cpu/compute.cpp index 097b4bd7e1..319c0acfcd 100644 --- a/src/libtorchaudio/rnnt/cpu/compute.cpp +++ b/src/libtorchaudio/rnnt/cpu/compute.cpp @@ -117,7 +117,7 @@ std::tuple> compute( /*logits=*/logits.data_ptr(), /*targets=*/targets.data_ptr(), /*logit_lengths=*/logit_lengths.data_ptr(), - /*target_lengths=*/target_lengths.data_ptr(), + /*tgtLengths=*/target_lengths.data_ptr(), /*costs=*/costs.data_ptr(), /*gradients=*/gradients->data_ptr()); break; @@ -128,7 +128,7 @@ std::tuple> compute( /*logits=*/logits.data_ptr(), /*targets=*/targets.data_ptr(), /*logit_lengths=*/logit_lengths.data_ptr(), - /*target_lengths=*/target_lengths.data_ptr(), + /*tgtLengths=*/target_lengths.data_ptr(), /*costs=*/costs.data_ptr(), /*gradients=*/gradients->data_ptr()); break; diff --git a/src/libtorchaudio/rnnt/cpu/compute_alphas.cpp b/src/libtorchaudio/rnnt/cpu/compute_alphas.cpp index 6923cbe5d8..6a11f81aee 100644 --- a/src/libtorchaudio/rnnt/cpu/compute_alphas.cpp +++ b/src/libtorchaudio/rnnt/cpu/compute_alphas.cpp @@ -56,7 +56,7 @@ torch::Tensor compute_alphas( /*logits=*/logits.data_ptr(), /*targets=*/targets.data_ptr(), /*logit_lengths=*/logit_lengths.data_ptr(), - /*target_lengths=*/target_lengths.data_ptr(), + /*tgtLengths=*/target_lengths.data_ptr(), /*alphas=*/alphas.data_ptr()); return alphas; } diff --git a/src/libtorchaudio/rnnt/cpu/compute_betas.cpp b/src/libtorchaudio/rnnt/cpu/compute_betas.cpp index d812ef34c3..f6a9f083ce 100644 --- a/src/libtorchaudio/rnnt/cpu/compute_betas.cpp +++ b/src/libtorchaudio/rnnt/cpu/compute_betas.cpp @@ -60,7 +60,7 @@ torch::Tensor compute_betas( /*logits=*/logits.data_ptr(), /*targets=*/targets.data_ptr(), /*logit_lengths=*/logit_lengths.data_ptr(), - /*target_lengths=*/target_lengths.data_ptr(), + /*tgtLengths=*/target_lengths.data_ptr(), /*costs=*/costs.data_ptr(), /*betas=*/betas.data_ptr()); return betas;