refine code

revert-15774-anakin_subgraph_engine
xuezhong 7 years ago
parent 4c98c2ccc3
commit 15d52f09f3

@ -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),

Loading…
Cancel
Save