(self)
| 735 | self.assertAllEqual(np_max, tf_max) |
| 736 | |
| 737 | def testBasic(self): |
| 738 | x = np.random.rand(1, 3, 2) * 100. |
| 739 | y = np.random.rand(1, 3, 2) * 100. |
| 740 | for t in [np.float16, np.float32, np.float64, np.int32, np.int64]: |
| 741 | self._compare(x.astype(t), y.astype(t), use_gpu=False) |
| 742 | self._compare(x.astype(t), y.astype(t), use_gpu=True) |
| 743 | |
| 744 | def testDifferentShapes(self): |
| 745 | x = np.random.rand(1, 3, 2) * 100. |