aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/layers
diff options
context:
space:
mode:
authorGravatar josephyearsley <joggino23@gmail.com>2018-07-23 23:06:48 +0100
committerGravatar Joe Yearsley <josephelliotyearsley@gmail.com>2018-09-29 17:31:57 +0100
commit46fc7a9530e9c8f6bf909de8df8c97e4b38a99a5 (patch)
tree2e0ff5b921c51a8a5db378d65aaafa71f2ff75a2 /tensorflow/python/layers
parent4de591a03a9bd49a05d67fe48f9358dbdac51561 (diff)
Fixed Tests
Diffstat (limited to 'tensorflow/python/layers')
-rw-r--r--tensorflow/python/layers/core_test.py4
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')