from .non_local import NonLocal2D
from .generalized_attention import GeneralizedAttention

__all__ = ['NonLocal2D', 'GeneralizedAttention']