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

Allow AUTOTUNE to be an acceptable value for num_parallel_calls.

Browse Source 
PiperOrigin-RevId: 254252931
This commit is contained in:
A. Unique TensorFlower 2019-06-20 12:49:57 -07:00 committed by TensorFlower Gardener
parent 889c063f0e
commit c759a9fb39
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/core/kernels/data/experimental/parse_example_dataset_op.cc
View File

@ -75,9 +75,10 @@ class ParseExampleDatasetOp : public UnaryDatasetOpKernel {
int64 num_parallel_calls = 0;
OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "num_parallel_calls",
&num_parallel_calls));
OP_REQUIRES(ctx, num_parallel_calls > 0,
errors::InvalidArgument(
"num_parallel_calls must be greater than zero."));
OP_REQUIRES(
ctx, num_parallel_calls > 0 || num_parallel_calls == model::kAutoTune,
errors::InvalidArgument(
"num_parallel_calls must be greater than zero."));
OpInputList dense_default_tensors;
OP_REQUIRES_OK(ctx,