diff --git a/keras_contrib/backend/tensorflow_backend.py b/keras_contrib/backend/tensorflow_backend.py index 008e876..aef687c 100644 --- a/keras_contrib/backend/tensorflow_backend.py +++ b/keras_contrib/backend/tensorflow_backend.py @@ -10,5 +10,5 @@ from keras import backend as K import numpy as np import os import warnings -from keras.backend import floatx, _EPSILON, image_dim_ordering, reset_uids +from keras.backend.common import floatx, _EPSILON, image_dim_ordering, reset_uids py_all = all diff --git a/keras_contrib/backend/theano_backend.py b/keras_contrib/backend/theano_backend.py index 42804e7..27db0fb 100644 --- a/keras_contrib/backend/theano_backend.py +++ b/keras_contrib/backend/theano_backend.py @@ -15,5 +15,5 @@ except ImportError: from keras import backend as K import inspect import numpy as np -from keras.backend import _FLOATX, floatx, _EPSILON, image_dim_ordering +from keras.backend.common import _FLOATX, floatx, _EPSILON, image_dim_ordering py_all = all