aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/keras/testing_utils.py
diff options
context:
space:
mode:
authorGravatar Francois Chollet <fchollet@google.com>2018-10-08 10:43:01 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-10-08 10:49:56 -0700
commit8ef3e7c8c053cb6dad530e13c478bbd406ea2c95 (patch)
tree74f36c8bd9293854ce0ee1f8a9bac04a863bfe99 /tensorflow/python/keras/testing_utils.py
parent153decedefc8da1fbd0717f4223b4b053e7aa517 (diff)
Part 1/3 of the feature sync to the Keras 2.2.4 API.
PiperOrigin-RevId: 216211279
Diffstat (limited to 'tensorflow/python/keras/testing_utils.py')
-rw-r--r--tensorflow/python/keras/testing_utils.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/python/keras/testing_utils.py b/tensorflow/python/keras/testing_utils.py
index 501b50ba5f..2fae094a1e 100644
--- a/tensorflow/python/keras/testing_utils.py
+++ b/tensorflow/python/keras/testing_utils.py
@@ -166,8 +166,9 @@ def layer_test(layer_cls, kwargs=None, input_shape=None, input_dtype=None,
if expected_dim is not None:
if expected_dim != actual_dim:
raise AssertionError(
- 'When testing layer %s, for input %s, found output_shape='
- '%s but expected to find %s.\nFull kwargs: %s' %
+ 'When testing layer %s **after deserialization**, '
+ 'for input %s, found output_shape='
+ '%s but expected to find inferred shape %s.\nFull kwargs: %s' %
(layer_cls.__name__,
x,
actual_output_shape,