[codemod] Fix argument comment issue in pytorch/audio/src/libtorchaudio/rnnt/cpu/compute.cpp +6

Differential Revision: D78633890

Pull Request resolved: #3987

Author: r-barnes

src/libtorchaudio/rnnt/cpu/compute.cpp

@@ -116,8 +116,8 @@ std::tuple<torch::Tensor, std::optional<torch::Tensor>> compute(
           /*workspace=*/workspace,
           /*logits=*/logits.data_ptr<float>(),
           /*targets=*/targets.data_ptr<int>(),
-          /*logit_lengths=*/logit_lengths.data_ptr<int>(),
-          /*target_lengths=*/target_lengths.data_ptr<int>(),
+          /*srcLengths=*/logit_lengths.data_ptr<int>(),
+          /*tgtLengths=*/target_lengths.data_ptr<int>(),
           /*costs=*/costs.data_ptr<float>(),
           /*gradients=*/gradients->data_ptr<float>());
       break;
@@ -127,8 +127,8 @@ std::tuple<torch::Tensor, std::optional<torch::Tensor>> compute(
           /*workspace=*/workspace,
           /*logits=*/logits.data_ptr<c10::Half>(),
           /*targets=*/targets.data_ptr<int>(),
-          /*logit_lengths=*/logit_lengths.data_ptr<int>(),
-          /*target_lengths=*/target_lengths.data_ptr<int>(),
+          /*srcLengths=*/logit_lengths.data_ptr<int>(),
+          /*tgtLengths=*/target_lengths.data_ptr<int>(),
           /*costs=*/costs.data_ptr<c10::Half>(),
           /*gradients=*/gradients->data_ptr<c10::Half>());
       break;

src/libtorchaudio/rnnt/cpu/compute_alphas.cpp

@@ -55,8 +55,8 @@ torch::Tensor compute_alphas(
       /*workspace=*/workspace,
       /*logits=*/logits.data_ptr<float>(),
       /*targets=*/targets.data_ptr<int>(),
-      /*logit_lengths=*/logit_lengths.data_ptr<int>(),
-      /*target_lengths=*/target_lengths.data_ptr<int>(),
+      /*srcLengths=*/logit_lengths.data_ptr<int>(),
+      /*tgtLengths=*/target_lengths.data_ptr<int>(),
       /*alphas=*/alphas.data_ptr<float>());
   return alphas;
 }

src/libtorchaudio/rnnt/cpu/compute_betas.cpp

@@ -59,8 +59,8 @@ torch::Tensor compute_betas(
       /*workspace=*/workspace,
       /*logits=*/logits.data_ptr<float>(),
       /*targets=*/targets.data_ptr<int>(),
-      /*logit_lengths=*/logit_lengths.data_ptr<int>(),
-      /*target_lengths=*/target_lengths.data_ptr<int>(),
+      /*srcLengths=*/logit_lengths.data_ptr<int>(),
+      /*tgtLengths=*/target_lengths.data_ptr<int>(),
       /*costs=*/costs.data_ptr<float>(),
       /*betas=*/betas.data_ptr<float>());
   return betas;