from .non_local import NonLocal2D
from .generalized_attention import GeneralizedAttention
__all__ = ['NonLocal2D', 'GeneralizedAttention']