diff options
author | 2017-09-13 12:17:42 -0700 | |
---|---|---|
committer | 2017-09-13 12:21:58 -0700 | |
commit | e74e16b3f78d1831f15259cb4c5cd7e594dcd3b8 (patch) | |
tree | 62a44146f0a8815f015e7770136288154c4ecc26 /tensorflow/python/training/training_ops.py | |
parent | 2dd412ffd834f447c80aee7308eb0ab520c237b9 (diff) |
Refactor optimizers and cleanup registration
PiperOrigin-RevId: 168576795
Diffstat (limited to 'tensorflow/python/training/training_ops.py')
-rw-r--r-- | tensorflow/python/training/training_ops.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/tensorflow/python/training/training_ops.py b/tensorflow/python/training/training_ops.py index e98c32b614..b33e7cb274 100644 --- a/tensorflow/python/training/training_ops.py +++ b/tensorflow/python/training/training_ops.py @@ -19,8 +19,16 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function -from tensorflow.python.training import gen_training_ops # go/tf-wildcard-import # pylint: disable=wildcard-import -from tensorflow.python.training.gen_training_ops import * +from tensorflow.python.training.gen_adadelta_ops import * +from tensorflow.python.training.gen_adagrad_da_ops import * +from tensorflow.python.training.gen_adagrad_ops import * +from tensorflow.python.training.gen_adam_ops import * +from tensorflow.python.training.gen_ftrl_ops import * +from tensorflow.python.training.gen_gradient_descent_ops import * +from tensorflow.python.training.gen_momentum_ops import * +from tensorflow.python.training.gen_proximal_adagrad_ops import * +from tensorflow.python.training.gen_proximal_gradient_descent_ops import * +from tensorflow.python.training.gen_rms_prop_ops import * # pylint: enable=wildcard-import |