diff options
-rw-r--r-- | tensorflow/core/kernels/fixed_length_record_reader_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/fixed_length_record_reader_op.cc b/tensorflow/core/kernels/fixed_length_record_reader_op.cc index 298a4e3678..bd2e28cefa 100644 --- a/tensorflow/core/kernels/fixed_length_record_reader_op.cc +++ b/tensorflow/core/kernels/fixed_length_record_reader_op.cc @@ -183,7 +183,7 @@ class FixedLengthRecordReaderOp : public ReaderOpKernel { errors::InvalidArgument("hop_bytes must be >= 0 not ", hop_bytes)); Env* env = context->env(); string encoding; - TF_CHECK_OK(context->GetAttr("encoding", &encoding)); + OP_REQUIRES_OK(context, context->GetAttr("encoding", &encoding)); SetReaderFactory([this, header_bytes, record_bytes, footer_bytes, hop_bytes, encoding, env]() { return new FixedLengthRecordReader(name(), header_bytes, record_bytes, |