Merge pull request #40284 from ROCmSoftwarePlatform:google_upstream_rocm_platform_fix_200608

PiperOrigin-RevId: 315492021
Change-Id: I3ff1493149b8a16786bb3a342198c50768e3795f
This commit is contained in:
TensorFlower Gardener 2020-06-09 08:57:05 -07:00
commit a866c95a1a

View File

@ -159,7 +159,7 @@ class MathTest(test.TestCase, parameterized.TestCase):
actual.shape, expected.shape,
'Shape mismatch.\nActual: {}\nExpected: {}\n{}'.format(
actual.shape, expected.shape, msg))
np.testing.assert_almost_equal(actual.tolist(), expected.tolist())
np.testing.assert_allclose(actual.tolist(), expected.tolist(), rtol=1e-6)
def testArgsort(self):
self._testUnaryOp(np_math_ops.argsort, np.argsort, 'argsort')