Migrate ops for new version of TensorForest.
PiperOrigin-RevId: 159852889
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD
index 6213ff6..42f7ac7 100755
--- a/tensorflow/contrib/BUILD
+++ b/tensorflow/contrib/BUILD
@@ -91,6 +91,8 @@
"//tensorflow/contrib/layers:sparse_feature_cross_op_kernel",
"//tensorflow/contrib/nccl:nccl_kernels",
"//tensorflow/contrib/seq2seq:beam_search_ops_kernels",
+ "//tensorflow/contrib/tensor_forest:model_ops_kernels",
+ "//tensorflow/contrib/tensor_forest:stats_ops_kernels",
"//tensorflow/contrib/tensor_forest:tensor_forest_kernels",
"//tensorflow/contrib/text:all_kernels",
],
@@ -108,6 +110,8 @@
"//tensorflow/contrib/layers:sparse_feature_cross_op_op_lib",
"//tensorflow/contrib/nccl:nccl_ops_op_lib",
"//tensorflow/contrib/seq2seq:beam_search_ops_op_lib",
+ "//tensorflow/contrib/tensor_forest:model_ops_op_lib",
+ "//tensorflow/contrib/tensor_forest:stats_ops_op_lib",
"//tensorflow/contrib/tensor_forest:tensor_forest_ops_op_lib",
"//tensorflow/contrib/text:all_ops",
"//tensorflow/contrib/tpu:all_ops",