aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/keras/models_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/python/keras/models_test.py')
-rw-r--r--tensorflow/python/keras/models_test.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/tensorflow/python/keras/models_test.py b/tensorflow/python/keras/models_test.py
index c616d8f24f..e6e45902a8 100644
--- a/tensorflow/python/keras/models_test.py
+++ b/tensorflow/python/keras/models_test.py
@@ -144,5 +144,19 @@ class CheckpointingTests(test.TestCase):
model.load_weights(save_prefix)
self.assertEqual(12., self.evaluate(beta1_power))
+class TestModelBackend(test.TestCase):
+
+ def test_model_backend_float64_use_cases(self):
+ # Test case for GitHub issue 19318
+ floatx = keras.backend.floatx()
+ keras.backend.set_floatx('float64')
+
+ x = keras.Input((5,))
+ y = keras.layers.Dense(1)(x)
+ model = keras.models.Model(x, y)
+ model.compile('rmsprop', 'mse')
+
+ keras.backend.set_floatx(floatx)
+
if __name__ == '__main__':
test.main()