diff options
Diffstat (limited to 'tensorflow/core/kernels/boosted_trees/BUILD')
-rw-r--r-- | tensorflow/core/kernels/boosted_trees/BUILD | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/boosted_trees/BUILD b/tensorflow/core/kernels/boosted_trees/BUILD index 62327dfe1d..4910021c63 100644 --- a/tensorflow/core/kernels/boosted_trees/BUILD +++ b/tensorflow/core/kernels/boosted_trees/BUILD @@ -30,6 +30,7 @@ tf_kernel_library( "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:lib_internal", + "//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_cc", ], ) @@ -44,6 +45,11 @@ cc_library( ], ) +cc_library( + name = "tree_helper", + hdrs = ["tree_helper.h"], +) + tf_kernel_library( name = "resource_ops", srcs = ["resource_ops.cc"], @@ -60,6 +66,7 @@ tf_kernel_library( name = "stats_ops", srcs = ["stats_ops.cc"], deps = [ + ":tree_helper", "//tensorflow/core:boosted_trees_ops_op_lib", "//tensorflow/core:framework", "//tensorflow/core:lib", @@ -71,6 +78,7 @@ tf_kernel_library( srcs = ["training_ops.cc"], deps = [ ":resources", + ":tree_helper", "//tensorflow/core:boosted_trees_ops_op_lib", "//tensorflow/core:framework", "//tensorflow/core:lib", |