__author__ = "<NAME>"

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

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

__author__ = "<NAME>"

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

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