diff --git a/keras_contrib/backend/__init__.py b/keras_contrib/backend/__init__.py index d903417..9b93e9b 100644 --- a/keras_contrib/backend/__init__.py +++ b/keras_contrib/backend/__init__.py @@ -12,3 +12,5 @@ if K.backend() == 'theano': from .theano_backend import * elif K.backend() == 'tensorflow': from .tensorflow_backend import * +elif K.backend() == 'cntk': + from .cntk_backend import * diff --git a/keras_contrib/backend/cntk_backend.py b/keras_contrib/backend/cntk_backend.py new file mode 100644 index 0000000..363ad08 --- /dev/null +++ b/keras_contrib/backend/cntk_backend.py @@ -0,0 +1,2 @@ +from keras.backend import cntk_backend as KCN +import cntk as C