__init__.py

from keras import backend as K

# We import all keras backend functions here,
# so that files in this repo can import both
# core and contrib backend functions with a
# single import statement.

if K.backend() == 'theano':
    from .theano_backend import *
elif K.backend() == 'tensorflow':
    from .tensorflow_backend import *
elif K.backend() == 'cntk':
    from .cntk_backend import *