aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/java/BUILD
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/java/BUILD
parent2dd412ffd834f447c80aee7308eb0ab520c237b9 (diff)
Refactor optimizers and cleanup registration
PiperOrigin-RevId: 168576795
Diffstat (limited to 'tensorflow/java/BUILD')
-rw-r--r--tensorflow/java/BUILD11
1 files changed, 10 insertions, 1 deletions
diff --git a/tensorflow/java/BUILD b/tensorflow/java/BUILD
index 4680e3ba16..26ece1d741 100644
--- a/tensorflow/java/BUILD
+++ b/tensorflow/java/BUILD
@@ -65,7 +65,16 @@ tf_java_op_gen_srcjar(
"sparse_ops",
"state_ops",
"string_ops",
- "training_ops",
+ "adadelta_ops",
+ "adagrad_da_ops",
+ "adagrad_ops",
+ "adam_ops",
+ "ftrl_ops",
+ "momentum_ops",
+ "gradient_descent_ops",
+ "proximal_adagrad_ops",
+ "proximal_gradient_descent_ops",
+ "rms_prop_ops",
"user_ops",
],
)