diff options
Diffstat (limited to 'tensorflow/contrib/opt/__init__.py')
-rw-r--r-- | tensorflow/contrib/opt/__init__.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/opt/__init__.py b/tensorflow/contrib/opt/__init__.py index 656a548cfd..f4cb7456cc 100644 --- a/tensorflow/contrib/opt/__init__.py +++ b/tensorflow/contrib/opt/__init__.py @@ -19,9 +19,11 @@ from __future__ import division from __future__ import print_function # pylint: disable=wildcard-import +from tensorflow.contrib.opt.python.training.delay_compensated_gradient_descent import * from tensorflow.contrib.opt.python.training.drop_stale_gradient_optimizer import * from tensorflow.contrib.opt.python.training.external_optimizer import * from tensorflow.contrib.opt.python.training.lazy_adam_optimizer import * +from tensorflow.contrib.opt.python.training.nadam_optimizer import * from tensorflow.contrib.opt.python.training.moving_average_optimizer import * from tensorflow.contrib.opt.python.training.nadam_optimizer import * from tensorflow.contrib.opt.python.training.variable_clipping_optimizer import * @@ -29,7 +31,9 @@ from tensorflow.contrib.opt.python.training.variable_clipping_optimizer import * from tensorflow.python.util.all_util import remove_undocumented + _allowed_symbols = [ + 'DelayCompensatedGradientDescentOptimizer', 'DropStaleGradientOptimizer', 'ExternalOptimizerInterface', 'LazyAdamOptimizer', 'NadamOptimizer', 'MovingAverageOptimizer', 'ScipyOptimizerInterface', 'VariableClippingOptimizer' |