From 48fa7d98814465c9c75f9d136766dc488bac3a1f Mon Sep 17 00:00:00 2001 From: Saurabh Saxena Date: Wed, 14 Aug 2019 16:03:32 -0700 Subject: [PATCH] Prepare cudnn_recurrent_test for Tensor equality. PiperOrigin-RevId: 263449689 (cherry picked from commit ff61cee968399a527f4ed57c622ae8dc60901f2d) --- tensorflow/python/keras/layers/cudnn_recurrent_test.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tensorflow/python/keras/layers/cudnn_recurrent_test.py b/tensorflow/python/keras/layers/cudnn_recurrent_test.py index bd266b5b282..3f0c72b5245 100644 --- a/tensorflow/python/keras/layers/cudnn_recurrent_test.py +++ b/tensorflow/python/keras/layers/cudnn_recurrent_test.py @@ -139,7 +139,9 @@ class CuDNNTest(keras_parameterized.TestCase): output = layer(inputs, initial_state=initial_state[0]) else: output = layer(inputs, initial_state=initial_state) - self.assertIn(initial_state[0], layer._inbound_nodes[0].input_tensors) + self.assertTrue( + any(initial_state[0] is t + for t in layer._inbound_nodes[0].input_tensors)) model = keras.models.Model([inputs] + initial_state, output) model.compile(