|
|
|
@ -350,7 +350,6 @@ class OpTest(unittest.TestCase):
|
|
|
|
actual_t = np.array(actual)
|
|
|
|
actual_t = np.array(actual)
|
|
|
|
expect = self.outputs[out_name]
|
|
|
|
expect = self.outputs[out_name]
|
|
|
|
expect_t = expect[0] if isinstance(expect, tuple) else expect
|
|
|
|
expect_t = expect[0] if isinstance(expect, tuple) else expect
|
|
|
|
#import pdb; pdb.set_trace()
|
|
|
|
|
|
|
|
self.assertTrue(
|
|
|
|
self.assertTrue(
|
|
|
|
np.allclose(
|
|
|
|
np.allclose(
|
|
|
|
actual_t, expect_t, atol=atol, equal_nan=equal_nan),
|
|
|
|
actual_t, expect_t, atol=atol, equal_nan=equal_nan),
|
|
|
|
|