commit 13e3719222f93e26b884b5d723eb6205a046bad9
author Scott Zhu <scottzhu@google.com> Sat Apr 04 23:27:22 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Sat Apr 04 23:33:41 2020 -0700
tree 5e6131646880885d8ae8f1a4d941f3a4b723f507
parent 21bbc9b07e1db33a4062b78c59da5793d2099013

Update ctc_loss_op_test to not rely on Keras.

PiperOrigin-RevId: 304855613
Change-Id: I99333fdf7e52e483d3742a160b6e3e6e64a8f5c6

tensorflow/python/kernel_tests/ctc_loss_op_test.py

diff --git a/tensorflow/python/kernel_tests/ctc_loss_op_test.py b/tensorflow/python/kernel_tests/ctc_loss_op_test.py
index 1991849..058437c 100644
--- a/tensorflow/python/kernel_tests/ctc_loss_op_test.py
+++ b/tensorflow/python/kernel_tests/ctc_loss_op_test.py
@@ -31,7 +31,6 @@
 from tensorflow.python.framework import random_seed
 from tensorflow.python.framework import sparse_tensor
 from tensorflow.python.framework import test_util
-from tensorflow.python.keras import keras_parameterized
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import ctc_ops
 from tensorflow.python.ops import gradients_impl
@@ -942,8 +941,8 @@
           [[1.0, 2.0], [5.0, 8.0], [14.0, 20.0]], out)
 
 
-@keras_parameterized.run_all_keras_modes
-class CTCLossTestV3(keras_parameterized.TestCase):
+@test_util.run_all_in_graph_and_eager_modes
+class CTCLossTestV3(test.TestCase, parameterized.TestCase):
 
   @parameterized.parameters([False, True])
   @test_util.run_v2_only
@@ -955,6 +954,8 @@
     """
     if not test.is_gpu_available():
       self.skipTest("Need GPU for testing.")
+    if not context.executing_eagerly():
+      self.skipTest("Need eager execution for testing.")
     random_seed.set_random_seed(5)
 
     batch_size = 8