def __new__(cls, *args, **kwargs):
        """ Creates attributes for the flatten and unflatten functionality. """
        obj = super().__new__(cls)
        obj.device = find_device([args, kwargs])
        return obj
 Creates attributes for the flatten and unflatten functionality. 
cola/ops/operator_base.py