@parameterized.named_parameters(itertools.chain.from_iterable((jtu.cases_from_list(({'testcase_name': '_{}_{}'.format(rec.op.__name__, special_value), 'op': rec.op, 'special_value': special_value, 'order': rec.order} for special_value in rec.values)) for rec in GRAD_SPECIAL_VALUE_TEST_RECORDS)))
def testOpGradSpecialValue(self, op, special_value, order):
    check_grads(op, (special_value,), order, ['fwd', 'rev'], atol={onp.float32: 0.003})
@parameterized.named_parameters(itertools.chain.from_iterable((jtu.cases_from_list(({'testcase_name': '_{}_{}'.format(rec.op.__name__, special_value), 'op': rec.op, 'special_value': special_value, 'order': rec.order} for special_value in rec.values)) for rec in GRAD_SPECIAL_VALUE_TEST_RECORDS)))
def testOpGradSpecialValue(self, op, special_value, order):
    check_grads(op, (special_value,), order, ['fwd', 'rev'], atol={onp.float32: 0.003})