diff options
Diffstat (limited to 'tensorflow/contrib/keras/python/keras/testing_utils.py')
-rw-r--r-- | tensorflow/contrib/keras/python/keras/testing_utils.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/keras/python/keras/testing_utils.py b/tensorflow/contrib/keras/python/keras/testing_utils.py index bf6f661adf..2f51ace945 100644 --- a/tensorflow/contrib/keras/python/keras/testing_utils.py +++ b/tensorflow/contrib/keras/python/keras/testing_utils.py @@ -78,7 +78,7 @@ def layer_test(layer_cls, kwargs=None, input_shape=None, input_dtype=None, if e is None: input_data_shape[i] = np.random.randint(1, 4) input_data = 10 * np.random.random(input_data_shape) - if input_dtype[:4] == 'float': + if input_dtype[:5] == 'float': input_data -= 0.5 input_data = input_data.astype(input_dtype) elif input_shape is None: |