diff options
Diffstat (limited to 'tensorflow/core/kernels/boosted_trees/BUILD')
-rw-r--r-- | tensorflow/core/kernels/boosted_trees/BUILD | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/boosted_trees/BUILD b/tensorflow/core/kernels/boosted_trees/BUILD index 4910021c63..4e8bfa02fc 100644 --- a/tensorflow/core/kernels/boosted_trees/BUILD +++ b/tensorflow/core/kernels/boosted_trees/BUILD @@ -15,7 +15,9 @@ load( tf_proto_library( name = "boosted_trees_proto", - srcs = ["boosted_trees.proto"], + srcs = [ + "boosted_trees.proto", + ], cc_api_version = 2, visibility = ["//visibility:public"], ) @@ -87,9 +89,21 @@ tf_kernel_library( ) tf_kernel_library( + name = "quantile_ops", + srcs = ["quantile_ops.cc"], + deps = [ + "//tensorflow/core:boosted_trees_ops_op_lib", + "//tensorflow/core:framework", + "//tensorflow/core:lib", + "//tensorflow/core/kernels/boosted_trees/quantiles:weighted_quantiles", + ], +) + +tf_kernel_library( name = "boosted_trees_ops", deps = [ ":prediction_ops", + ":quantile_ops", ":resource_ops", ":stats_ops", ":training_ops", |