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

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

@ -21,7 +21,7 @@ from __future__ import print_function
import numpy as np import numpy as np
from tensorflow.python.framework import dtypes 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 array_ops
from tensorflow.python.ops import parsing_ops from tensorflow.python.ops import parsing_ops
from tensorflow.python.platform import test from tensorflow.python.platform import test
@ -29,8 +29,9 @@ from tensorflow.python.platform import test
class DecodeRawOpTest(test.TestCase): class DecodeRawOpTest(test.TestCase):
@test_util.deprecated_graph_mode_only
def testShapeInference(self): def testShapeInference(self):
# Shape function requires placeholders and a graph.
with ops.Graph().as_default():
for dtype in [dtypes.bool, dtypes.int8, dtypes.uint8, dtypes.int16, for dtype in [dtypes.bool, dtypes.int8, dtypes.uint8, dtypes.int16,
dtypes.uint16, dtypes.int32, dtypes.int64, dtypes.float16, dtypes.uint16, dtypes.int32, dtypes.int64, dtypes.float16,
dtypes.float32, dtypes.float64, dtypes.complex64, dtypes.float32, dtypes.float64, dtypes.complex64,