aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/layers
diff options
context:
space:
mode:
authorGravatar josephyearsley <joggino23@gmail.com>2018-03-18 20:08:59 +0000
committerGravatar Joe Yearsley <josephelliotyearsley@gmail.com>2018-09-29 17:31:57 +0100
commit579aecd2de1f0582858f83e3c8da2a8dbb57993b (patch)
tree475743e3109d5d63e112cf178f129690b090ab81 /tensorflow/python/layers
parentdd928d5ae31dd0484e5e4a96c6322adecc4e511b (diff)
added dtype to test
Diffstat (limited to 'tensorflow/python/layers')
-rw-r--r--tensorflow/python/layers/core_test.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/layers/core_test.py b/tensorflow/python/layers/core_test.py
index 0d019897aa..31f3a4e0b0 100644
--- a/tensorflow/python/layers/core_test.py
+++ b/tensorflow/python/layers/core_test.py
@@ -477,7 +477,7 @@ class FlattenTest(test.TestCase):
self.assertEqual(shape.as_list(), [None, None])
def testDataFormat(self):
- np_input_channels_last = np.arange(3, 7).reshape([1, 2, 3, 2])
+ np_input_channels_last = np.arange(12, dtype='float32').reshape([1, 2, 3, 2])
with self.test_session() as sess:
x = array_ops.placeholder(shape=(1, 2, 3, 2), dtype='float32')