experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change the maximum input for model::MakeParameter::parallelism

Browse Source
This commit is contained in:
Fei Hu 2018-11-05 16:05:42 -08:00
parent 0ff931a596
commit 689d9974b9
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/data/experimental/numa_map_and_batch_dataset_op.cc
View File

@ -244,7 +244,7 @@ class NumaMapAndBatchDatasetOp : public UnaryDatasetOpKernel {
return model::MakeAsyncKnownRatioNode(
std::move(args), dataset()->batch_size_,
{model::MakeParameter("parallelism", num_parallel_calls_, /*min=*/1,
/*max=*/port::NumSchedulableCPUs())});
/*max=*/ctx->runner_threadpool_size())});
}
Status SaveInternal(IteratorStateWriter* writer) override {

2
tensorflow/core/kernels/data/map_and_batch_dataset_op.cc
View File

@ -296,7 +296,7 @@ class MapAndBatchDatasetOp : public UnaryDatasetOpKernel {
return model::MakeAsyncKnownRatioNode(
std::move(args), dataset()->batch_size_,
{model::MakeParameter("parallelism", num_parallel_calls_, /*min=*/1,
/*max=*/port::NumSchedulableCPUs())});
/*max=*/ctx->runner_threadpool_size())});
}
Status SaveInternal(IteratorStateWriter* writer) override {

2
tensorflow/core/kernels/data/parallel_map_iterator.cc
View File

@ -101,7 +101,7 @@ class ParallelMapIterator : public DatasetBaseIterator {
std::move(args),
/*ratio=*/1,
{model::MakeParameter("parallelism", num_parallel_calls_, /*min=*/1,
/*max=*/port::NumSchedulableCPUs())});
/*max=*/ctx->runner_threadpool_size())});
}
Status SaveInternal(IteratorStateWriter* writer) override {