diff --git a/tf_keras/__init__.py b/tf_keras/__init__.py index 34719ad43..326cb731e 100644 --- a/tf_keras/__init__.py +++ b/tf_keras/__init__.py @@ -24,6 +24,7 @@ from tf_keras import losses from tf_keras import metrics from tf_keras import models +from tf_keras import optimizers from tf_keras.engine.input_layer import Input from tf_keras.engine.sequential import Sequential from tf_keras.engine.training import Model