diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-12 10:47:26 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-12 10:50:24 -0700 |
commit | 151c31ce75f4370fd3749f3b07ac8297d3b2e203 (patch) | |
tree | e5d5fab632100e0dbdc7ddef8efa85db1805daeb /tensorflow/python/layers | |
parent | 844b8cae970d835850a75f8063324224b2de0df0 (diff) |
Make default weights initializer in `base_layers.Layer` suitable for their dtype.
PiperOrigin-RevId: 192634133
Diffstat (limited to 'tensorflow/python/layers')
-rw-r--r-- | tensorflow/python/layers/base_test.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/python/layers/base_test.py b/tensorflow/python/layers/base_test.py index c05c675263..f08b552840 100644 --- a/tensorflow/python/layers/base_test.py +++ b/tensorflow/python/layers/base_test.py @@ -53,6 +53,12 @@ class BaseLayerTest(test.TestCase): self.assertEqual(layer.trainable, False) @test_util.run_in_graph_and_eager_modes() + def testInt64Layer(self): + layer = base_layers.Layer(name='my_layer', dtype='int64') + layer.add_variable('my_var', [2, 2]) + self.assertEqual(layer.name, 'my_layer') + + @test_util.run_in_graph_and_eager_modes() def testAddWeight(self): layer = base_layers.Layer(name='my_layer') |