__author__ = "Alex Rogozhnikov"

from ..layers.tensorflow import Rearrange, Reduce, EinMix

keras_custom_objects = {
    Rearrange.__name__: Rearrange,
    Reduce.__name__: Reduce,
    EinMix.__name__: EinMix,
}