def METHOD_NAME(batch):
    """
    Move all values in a dictionary to cpu if supported.
    Args:
        batch: Input dict.
    Returns:
        batch_cpu: `batch` moved to a cpu device, if supported.
    """
    return {k: try_to_cpu(v) for k, v in batch.items()}