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

Eager execution coverage for decode_raw_op_test.py. Removed run_deprecated_v1 decorators.

Browse Source 
PiperOrigin-RevId: 327043724
Change-Id: I6af2449e89928009696ba01b8d42cf4aae06b021
This commit is contained in:
Hye Soo Yang 2020-08-17 10:07:48 -07:00 committed by TensorFlower Gardener
parent 91bcb88f53
commit c74b67b392
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tensorflow/python/kernel_tests/decode_raw_op_test.py
View File

@ -21,7 +21,7 @@ from __future__ import print_function
import numpy as np
from tensorflow.python.framework import dtypes
from tensorflow.python.framework import test_util
from tensorflow.python.framework import ops
from tensorflow.python.ops import array_ops
from tensorflow.python.ops import parsing_ops
from tensorflow.python.platform import test
@ -29,15 +29,16 @@ from tensorflow.python.platform import test
class DecodeRawOpTest(test.TestCase):
@test_util.deprecated_graph_mode_only
def testShapeInference(self):
for dtype in [dtypes.bool, dtypes.int8, dtypes.uint8, dtypes.int16,
dtypes.uint16, dtypes.int32, dtypes.int64, dtypes.float16,
dtypes.float32, dtypes.float64, dtypes.complex64,
dtypes.complex128]:
in_bytes = array_ops.placeholder(dtypes.string, shape=[None])
decode = parsing_ops.decode_raw(in_bytes, dtype)
self.assertEqual([None, None], decode.get_shape().as_list())
# Shape function requires placeholders and a graph.
with ops.Graph().as_default():
for dtype in [dtypes.bool, dtypes.int8, dtypes.uint8, dtypes.int16,
dtypes.uint16, dtypes.int32, dtypes.int64, dtypes.float16,
dtypes.float32, dtypes.float64, dtypes.complex64,
dtypes.complex128]:
in_bytes = array_ops.placeholder(dtypes.string, shape=[None])
decode = parsing_ops.decode_raw(in_bytes, dtype)
self.assertEqual([None, None], decode.get_shape().as_list())
def testToUint8(self):
self.assertAllEqual(