diff options
Diffstat (limited to 'tensorflow/core/kernels/boosted_trees/BUILD')
| -rw-r--r-- | tensorflow/core/kernels/boosted_trees/BUILD | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/boosted_trees/BUILD b/tensorflow/core/kernels/boosted_trees/BUILD index 0244f3cd8d..4910021c63 100644 --- a/tensorflow/core/kernels/boosted_trees/BUILD +++ b/tensorflow/core/kernels/boosted_trees/BUILD @@ -45,6 +45,11 @@ cc_library( ], ) +cc_library( + name = "tree_helper", + hdrs = ["tree_helper.h"], +) + tf_kernel_library( name = "resource_ops", srcs = ["resource_ops.cc"], @@ -61,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", @@ -72,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", |