def setUp(self):
    self._target_ops = {'elementwise_add', 'pool2d'}
    self._target_grad_ops = {'elementwise_add_grad', 'pool2d_grad'}
def setUp(self):
    self._target_ops = {'elementwise_add', 'pool2d'}
    self._target_grad_ops = {'elementwise_add_grad', 'pool2d_grad'}