licenses(["notice"])  # Apache 2.0

exports_files(["LICENSE"])

package(
    default_visibility = [
        "//tensorflow/contrib/boosted_trees:__subpackages__",
        "//tensorflow/contrib/boosted_trees:friends",
    ],
)

filegroup(
    name = "all_files",
    srcs = glob(
        ["**/*"],
        exclude = [
            "**/OWNERS",
        ],
    ),
    visibility = ["//tensorflow:__subpackages__"],
)

cc_library(
    name = "stamped_resource",
    hdrs = ["stamped_resource.h"],
    deps = [
        "//tensorflow/core:framework_headers_lib",
        "//third_party/eigen3",
    ],
)

cc_library(
    name = "quantile_stream_resource",
    hdrs = ["quantile_stream_resource.h"],
    deps = [
        ":stamped_resource",
        "//tensorflow/contrib/boosted_trees/lib:weighted_quantiles",
        "//tensorflow/contrib/boosted_trees/proto:quantiles_proto_cc",
        "//tensorflow/core:framework_headers_lib",
        "//third_party/eigen3",
    ],
)

cc_library(
    name = "decision_tree_ensemble_resource",
    hdrs = ["decision_tree_ensemble_resource.h"],
    deps = [
        ":stamped_resource",
        "//tensorflow/contrib/boosted_trees/lib:trees",
        "//tensorflow/core:framework_headers_lib",
    ],
    alwayslink = 1,
)
