@@ -84,9 +84,9 @@ void CTCLoss::execute([[maybe_unused]] const dnnl::stream& strm) {
8484 int32_t returnCode = 0 ;
8585
8686 const auto * logits = getSrcDataAtPortAs<const float >(0 );
87- const auto * logitsLength = getSrcDataAtPortAs<const int32_t >(1 );
88- const auto * labels = getSrcDataAtPortAs<const int32_t >(2 );
89- const auto * labelsLength = getSrcDataAtPortAs<const int32_t >(3 );
87+ const auto * logitsLength = getSrcDataAtPortAs<const int >(1 );
88+ const auto * labels = getSrcDataAtPortAs<const int >(2 );
89+ const auto * labelsLength = getSrcDataAtPortAs<const int >(3 );
9090 auto * dstData = getDstDataAtPortAs<float >(0 );
9191
9292 const auto & inDims = getParentEdgeAt (0 )->getMemory ().getStaticDims ();
@@ -99,8 +99,8 @@ void CTCLoss::execute([[maybe_unused]] const dnnl::stream& strm) {
9999 blankIndex = getSrcDataAtPortAs<const int32_t >(4 )[0 ];
100100 }
101101
102- std::vector<int32_t > decodedTargetLenB (batchNum, 0 );
103- std::vector<std::vector<int32_t >> targetDB (batchNum);
102+ std::vector<int > decodedTargetLenB (batchNum, 0 );
103+ std::vector<std::vector<int >> targetDB (batchNum);
104104 std::vector<std::vector<std::vector<float >>> logProbabilitiesB (batchNum);
105105 const auto threads_num = parallel_get_max_threads ();
106106 std::vector<std::string> errorMsgB (threads_num);
@@ -268,16 +268,16 @@ void CTCLoss::execute([[maybe_unused]] const dnnl::stream& strm) {
268268 for (size_t b = start; b < end; b++) {
269269 auto & targetD = targetDB[b];
270270 auto & logProbabilities = logProbabilitiesB[b];
271- const int32_t actualLogitLen = logitsLength[b];
272- const int32_t decodedTargetLen = decodedTargetLenB[b];
271+ const int actualLogitLen = logitsLength[b];
272+ const int decodedTargetLen = decodedTargetLenB[b];
273273 std::vector<std::vector<float >> logBwd (decodedTargetLen, std::vector<float >(actualLogitLen, -float_inf));
274- for (int32_t s = decodedTargetLen - 2 ; s < decodedTargetLen; s++) {
274+ for (int s = decodedTargetLen - 2 ; s < decodedTargetLen; s++) {
275275 logBwd[s][actualLogitLen - 1 ] = 0 .F ;
276276 }
277277
278- for (int32_t t = actualLogitLen - 2 ; t >= 0 ; t--) {
279- const int32_t t_1 = t + 1 ;
280- for (int32_t s = std::max (0 , decodedTargetLen - (2 * (actualLogitLen - t)));
278+ for (int t = actualLogitLen - 2 ; t >= 0 ; t--) {
279+ const int t_1 = t + 1 ;
280+ for (int s = std::max (0 , decodedTargetLen - (2 * (actualLogitLen - t)));
281281 s < std::min (decodedTargetLen, 2 * (t_1));
282282 s++) {
283283 if (ctcMergeRepeated || targetD[s] == blankIndex) {
0 commit comments