diff options
author | 2018-10-08 10:43:01 -0700 | |
---|---|---|
committer | 2018-10-08 10:49:56 -0700 | |
commit | 8ef3e7c8c053cb6dad530e13c478bbd406ea2c95 (patch) | |
tree | 74f36c8bd9293854ce0ee1f8a9bac04a863bfe99 /tensorflow/python/keras/testing_utils.py | |
parent | 153decedefc8da1fbd0717f4223b4b053e7aa517 (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.py | 5 |
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, |