restore parallelization

microsoft · Nov 21, 2023 · bbb603c · bbb603c
1 parent 6d6c82c
commit bbb603c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/onnxruntime/core/providers/cpu/nn/tfidfvectorizer.cc b/onnxruntime/core/providers/cpu/nn/tfidfvectorizer.cc
@@ -428,8 +428,7 @@ Status TfIdfVectorizer::Compute(OpKernelContext* ctx) const {
     OutputResult(frequencies, output_data + row_num * this->impl_->output_size_);
   };
 
-  // concurrency::ThreadPool::TryBatchParallelFor(ctx->GetOperatorThreadPool(), num_rows, std::move(fn), 0);
-  for (size_t i = 0; i < num_rows; ++i) fn(i);
+  concurrency::ThreadPool::TryBatchParallelFor(ctx->GetOperatorThreadPool(), num_rows, std::move(fn), 0);
   return Status::OK();
 }