From 14d5ba1227d1d78ea9fe299db0cfacbbdb41970f Mon Sep 17 00:00:00 2001 From: Fariz Rahman Date: Sat, 10 Jun 2017 10:05:11 +0400 Subject: [PATCH] Add cntk backend (#103) * add cntk backend * reorder imports --- keras_contrib/backend/__init__.py | 2 ++ keras_contrib/backend/cntk_backend.py | 2 ++ 2 files changed, 4 insertions(+) create mode 100644 keras_contrib/backend/cntk_backend.py 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