def METHOD_NAME(device):
    inps = [torch.Tensor(x), torch.Tensor(y)]
    if device == "cpu":
        return inps
    else:
        device_type, device_id = device.split(":")
        assert device_type == "cuda"
        return [inp.cuda(int(device_id)) for inp in inps]