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

[tf.data] Fix some small typo in parallel_interleave_dataset_op.

Browse Source 
PiperOrigin-RevId: 316571471
Change-Id: Ie4b11e5256b88086a7efb3c87077936889a3fe10
This commit is contained in:
Jay Shi 2020-06-15 16:44:00 -07:00 committed by TensorFlower Gardener
parent bfbec5fb03
commit 7a43680c0e
2 changed files with 8 additions and 6 deletions

6
tensorflow/core/kernels/data/parallel_interleave_dataset_op.cc
View File

@ -1544,11 +1544,13 @@ void ParallelInterleaveDatasetOp::MakeDataset(OpKernelContext* ctx,
int64 buffer_output_elements = model::kAutotune;
int64 prefetch_input_elements = model::kAutotune;
if (op_version_ >= 4) {
OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kBufferOutputElements,
&buffer_output_elements));
OP_REQUIRES(ctx,
buffer_output_elements == model::kAutotune ||
buffer_output_elements >= 0,
buffer_output_elements > 0,
errors::InvalidArgument("`buffer_output_elements` must be ",
model::kAutotune, " or >= 0 but is ",
model::kAutotune, " or > 0 but is ",
buffer_output_elements));
OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kPrefetchInputElements,

8
tensorflow/core/kernels/data/parallel_interleave_dataset_op_test.cc
View File

@ -159,7 +159,7 @@ ParallelInterleaveDatasetParams ParallelInterleaveDatasetParams2() {
/*other_arguments=*/{},
/*cycle_length=*/2,
/*block_length=*/1,
/*buffer_output_elements=*/0,
/*buffer_output_elements=*/1,
/*prefetch_input_elements=*/0,
/*num_parallel_calls=*/2,
/*func=*/
@ -184,7 +184,7 @@ ParallelInterleaveDatasetParams ParallelInterleaveDatasetParams3() {
/*other_arguments=*/{},
/*cycle_length=*/3,
/*block_length=*/1,
/*buffer_output_elements=*/0,
/*buffer_output_elements=*/1,
/*prefetch_input_elements=*/1,
/*num_parallel_calls=*/2,
/*func=*/
@ -488,7 +488,7 @@ ParallelInterleaveDatasetParamsWithInvalidBufferOutputElements() {
/*other_arguments=*/{},
/*cycle_length=*/1,
/*block_length=*/1,
/*buffer_output_elements=*/-2,
/*buffer_output_elements=*/model::kAutotune,
/*prefetch_input_elements=*/model::kAutotune,
/*num_parallel_calls=*/-5,
/*func=*/
@ -514,7 +514,7 @@ ParallelInterleaveDatasetParamsWithInvalidPrefetchInputElements() {
/*other_arguments=*/{},
/*cycle_length=*/1,
/*block_length=*/1,
/*buffer_output_elements=*/-2,
/*buffer_output_elements=*/model::kAutotune,
/*prefetch_input_elements=*/model::kAutotune,
/*num_parallel_calls=*/-5,
/*func=*/