diff options
author | 2018-07-23 23:06:48 +0100 | |
---|---|---|
committer | 2018-09-29 17:31:57 +0100 | |
commit | 46fc7a9530e9c8f6bf909de8df8c97e4b38a99a5 (patch) | |
tree | 2e0ff5b921c51a8a5db378d65aaafa71f2ff75a2 /tensorflow/python/layers | |
parent | 4de591a03a9bd49a05d67fe48f9358dbdac51561 (diff) |
Fixed Tests
Diffstat (limited to 'tensorflow/python/layers')
-rw-r--r-- | tensorflow/python/layers/core_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/layers/core_test.py b/tensorflow/python/layers/core_test.py index 8ad0e8c4ba..22ed75dda7 100644 --- a/tensorflow/python/layers/core_test.py +++ b/tensorflow/python/layers/core_test.py @@ -491,7 +491,7 @@ class FlattenTest(test.TestCase): [0, 4, 1, 2, 3]) np_output_cf = sess.run(y, feed_dict={x: np_input_channels_first}) - self.assertEqual(np_output_cl, np_output_cf) + self.assertAllEqual(np_output_cl, np_output_cf) def testDataFormat4d(self): np_input_channels_last = np.arange(24, dtype='float32').reshape( @@ -508,7 +508,7 @@ class FlattenTest(test.TestCase): [0, 3, 1, 2]) np_output_cf = sess.run(y, feed_dict={x: np_input_channels_first}) - self.assertEqual(np_output_cl, np_output_cf) + self.assertAllEqual(np_output_cl, np_output_cf) def testFunctionalFlatten(self): x = array_ops.placeholder(shape=(None, 2, 3), dtype='float32') |