diff options
Diffstat (limited to 'tensorflow/python/ops/io_ops.py')
-rw-r--r-- | tensorflow/python/ops/io_ops.py | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/tensorflow/python/ops/io_ops.py b/tensorflow/python/ops/io_ops.py index 2c34c7ba27..5187242ebe 100644 --- a/tensorflow/python/ops/io_ops.py +++ b/tensorflow/python/ops/io_ops.py @@ -464,13 +464,11 @@ class TFRecordReader(ReaderBase): name: A name for the operation (optional). options: A TFRecordOptions object (optional). """ - compression_type_string = "" - if (options and - options.compression_type == python_io.TFRecordCompressionType.ZLIB): - compression_type_string = "ZLIB" + compression_type = python_io.TFRecordOptions.get_compression_type_string( + options) - rr = gen_io_ops._tf_record_reader(name=name, - compression_type=compression_type_string) + rr = gen_io_ops._tf_record_reader( + name=name, compression_type=compression_type) super(TFRecordReader, self).__init__(rr) |