85 def build_constraint(constraint_fn, args, is_cuda=False):  argument
88     t = torch.cuda.DoubleTensor if is_cuda else torch.DoubleTensor
94     "is_cuda",
102 def test_constraint(constraint_fn, result, value, is_cuda):  argument
103     t = torch.cuda.DoubleTensor if is_cuda else torch.DoubleTensor
111     "is_cuda",
119 def test_biject_to(constraint_fn, args, is_cuda):  argument
120     constraint = build_constraint(constraint_fn, args, is_cuda=is_cuda)
131     if is_cuda:
152     "is_cuda",
160 def test_transform_to(constraint_fn, args, is_cuda):  argument
161     constraint = build_constraint(constraint_fn, args, is_cuda=is_cuda)
168     if is_cuda: