mirror of
https://github.com/wassname/keras-contrib.git
synced 2026-07-19 11:24:38 +08:00
15 lines
359 B
Python
15 lines
359 B
Python
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.
|
|
|
|
from keras.backend import *
|
|
|
|
|
|
if K.backend() == 'theano':
|
|
from .theano_backend import *
|
|
elif K.backend() == 'tensorflow':
|
|
from .tensorflow_backend import *
|