Remove `no_grad_set` usage from "conv_shift" op tests (#22683)

revert-22710-feature/integrated_ps_api
Yang Zhang 5 years ago committed by GitHub
parent d39d8bee4e
commit 1a595d8e90
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -51,12 +51,10 @@ class TestConvShiftOp(OpTest):
self.check_grad(['X', 'Y'], 'Out')
def test_check_grad_ignore_x(self):
self.check_grad(
['Y'], 'Out', max_relative_error=0.05, no_grad_set=set("X"))
self.check_grad(['Y'], 'Out')
def test_check_grad_ignore_y(self):
self.check_grad(
['X'], 'Out', max_relative_error=0.05, no_grad_set=set('Y'))
self.check_grad(['X'], 'Out')
if __name__ == '__main__':

@ -23,7 +23,6 @@ NEED_TO_FIX_OP_LIST = [
'affine_grid',
'backward',
'batch_norm',
'conv_shift',
'conv2d',
'conv2d_transpose',
'conv3d',

Loading…
Cancel
Save