aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/training/training_ops.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-09-13 12:17:42 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-09-13 12:21:58 -0700
commite74e16b3f78d1831f15259cb4c5cd7e594dcd3b8 (patch)
tree62a44146f0a8815f015e7770136288154c4ecc26 /tensorflow/python/training/training_ops.py
parent2dd412ffd834f447c80aee7308eb0ab520c237b9 (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.py12
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