Add deprecation logging to boosted trees kernels.
These ops are no longer maintained and will be removed in the future.
PiperOrigin-RevId: 413751541
Change-Id: I3fdcb6489563b60b105c898e29c1162ad91ceb37
diff --git a/tensorflow/core/kernels/boosted_trees/prediction_ops.cc b/tensorflow/core/kernels/boosted_trees/prediction_ops.cc
index 831f6f2..b26d467 100644
--- a/tensorflow/core/kernels/boosted_trees/prediction_ops.cc
+++ b/tensorflow/core/kernels/boosted_trees/prediction_ops.cc
@@ -58,6 +58,9 @@
public:
explicit BoostedTreesTrainingPredictOp(OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr("num_bucketized_features",
&num_bucketized_features_));
OP_REQUIRES_OK(context,
@@ -201,6 +204,9 @@
public:
explicit BoostedTreesPredictOp(OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr("num_bucketized_features",
&num_bucketized_features_));
OP_REQUIRES_OK(context,
@@ -298,6 +304,9 @@
explicit BoostedTreesExampleDebugOutputsOp(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr("num_bucketized_features",
&num_bucketized_features_));
OP_REQUIRES_OK(context,
diff --git a/tensorflow/core/kernels/boosted_trees/quantile_ops.cc b/tensorflow/core/kernels/boosted_trees/quantile_ops.cc
index acf926f..5c23a4f 100644
--- a/tensorflow/core/kernels/boosted_trees/quantile_ops.cc
+++ b/tensorflow/core/kernels/boosted_trees/quantile_ops.cc
@@ -98,6 +98,9 @@
explicit BoostedTreesCreateQuantileStreamResourceOp(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr(kMaxElementsName, &max_elements_));
}
@@ -143,6 +146,9 @@
explicit BoostedTreesMakeQuantileSummariesOp(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr(kNumFeaturesName, &num_features_));
}
@@ -224,6 +230,9 @@
explicit BoostedTreesFlushQuantileSummariesOp(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr(kNumFeaturesName, &num_features_));
}
@@ -284,7 +293,11 @@
public:
explicit BoostedTreesQuantileStreamResourceAddSummariesOp(
OpKernelConstruction* const context)
- : OpKernel(context) {}
+ : OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
+ }
void Compute(OpKernelContext* context) override {
ResourceHandle handle;
@@ -348,6 +361,9 @@
explicit BoostedTreesQuantileStreamResourceDeserializeOp(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr(kNumStreamsName, &num_features_));
}
@@ -398,6 +414,9 @@
explicit BoostedTreesQuantileStreamResourceFlushOp(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context,
context->GetAttr(kGenerateQuantiles, &generate_quantiles_));
}
@@ -454,6 +473,9 @@
explicit BoostedTreesQuantileStreamResourceGetBucketBoundariesOp(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr(kNumFeaturesName, &num_features_));
}
@@ -512,6 +534,9 @@
public:
explicit BoostedTreesBucketizeOp(OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr(kNumFeaturesName, &num_features_));
}
diff --git a/tensorflow/core/kernels/boosted_trees/resource_ops.cc b/tensorflow/core/kernels/boosted_trees/resource_ops.cc
index 435c7d2..43f6662 100644
--- a/tensorflow/core/kernels/boosted_trees/resource_ops.cc
+++ b/tensorflow/core/kernels/boosted_trees/resource_ops.cc
@@ -36,7 +36,11 @@
class BoostedTreesCreateEnsembleOp : public OpKernel {
public:
explicit BoostedTreesCreateEnsembleOp(OpKernelConstruction* context)
- : OpKernel(context) {}
+ : OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
+ }
void Compute(OpKernelContext* context) override {
// Get the stamp token.
@@ -76,7 +80,11 @@
class BoostedTreesGetEnsembleStatesOp : public OpKernel {
public:
explicit BoostedTreesGetEnsembleStatesOp(OpKernelConstruction* context)
- : OpKernel(context) {}
+ : OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
+ }
void Compute(OpKernelContext* context) override {
// Looks up the resource.
@@ -139,7 +147,11 @@
class BoostedTreesSerializeEnsembleOp : public OpKernel {
public:
explicit BoostedTreesSerializeEnsembleOp(OpKernelConstruction* context)
- : OpKernel(context) {}
+ : OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
+ }
void Compute(OpKernelContext* context) override {
core::RefCountPtr<BoostedTreesEnsembleResource> tree_ensemble_resource;
@@ -166,7 +178,11 @@
class BoostedTreesDeserializeEnsembleOp : public OpKernel {
public:
explicit BoostedTreesDeserializeEnsembleOp(OpKernelConstruction* context)
- : OpKernel(context) {}
+ : OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
+ }
void Compute(OpKernelContext* context) override {
core::RefCountPtr<BoostedTreesEnsembleResource> tree_ensemble_resource;
diff --git a/tensorflow/core/kernels/boosted_trees/stats_ops.cc b/tensorflow/core/kernels/boosted_trees/stats_ops.cc
index 4583d45..103e4b2 100644
--- a/tensorflow/core/kernels/boosted_trees/stats_ops.cc
+++ b/tensorflow/core/kernels/boosted_trees/stats_ops.cc
@@ -45,6 +45,9 @@
explicit BoostedTreesCalculateBestGainsPerFeatureOp(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr("max_splits", &max_splits_));
OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_));
}
@@ -267,6 +270,9 @@
explicit BoostedTreesCalculateBestFeatureSplitOp(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr("logits_dimension", &logits_dim_));
OP_REQUIRES_OK(context, context->GetAttr("split_type", &split_type_));
}
@@ -622,6 +628,9 @@
explicit BoostedTreesCalculateBestFeatureSplitV2(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr("logits_dimension", &logits_dim_));
OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_));
}
@@ -1076,6 +1085,9 @@
explicit BoostedTreesSparseCalculateBestFeatureSplitOp(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
// TODO(crawles): Using logits_dim_ for multi-class split.
OP_REQUIRES_OK(context, context->GetAttr("logits_dimension", &logits_dim_));
// TODO(tanzheny): Using this for equality split.
@@ -1424,6 +1436,9 @@
public:
explicit BoostedTreesMakeStatsSummaryOp(OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr("max_splits", &max_splits_));
OP_REQUIRES_OK(context, context->GetAttr("num_buckets", &num_buckets_));
OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_));
@@ -1521,6 +1536,9 @@
public:
explicit BoostedTreesAggregateStatsOp(OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr("max_splits", &max_splits_));
OP_REQUIRES_OK(context, context->GetAttr("num_buckets", &num_buckets_));
}
@@ -1758,6 +1776,9 @@
explicit BoostedTreesSparseAggregateStatsOp(
OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr("max_splits", &max_splits_));
OP_REQUIRES_OK(context, context->GetAttr("num_buckets", &num_buckets_));
}
diff --git a/tensorflow/core/kernels/boosted_trees/training_ops.cc b/tensorflow/core/kernels/boosted_trees/training_ops.cc
index 3115d10..658854a 100644
--- a/tensorflow/core/kernels/boosted_trees/training_ops.cc
+++ b/tensorflow/core/kernels/boosted_trees/training_ops.cc
@@ -35,6 +35,9 @@
public:
explicit BoostedTreesUpdateEnsembleOp(OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_));
int32_t pruning_index;
@@ -234,6 +237,9 @@
public:
explicit BoostedTreesUpdateEnsembleV2Op(OpKernelConstruction* const context)
: OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
OP_REQUIRES_OK(context, context->GetAttr("logits_dimension", &logits_dim_));
OP_REQUIRES_OK(context, context->GetAttr("num_groups", &num_groups_));
}
@@ -457,7 +463,11 @@
class BoostedTreesCenterBiasOp : public OpKernel {
public:
explicit BoostedTreesCenterBiasOp(OpKernelConstruction* const context)
- : OpKernel(context) {}
+ : OpKernel(context) {
+ VLOG(1) << "Boosted Trees kernels in TF are deprecated. Please use "
+ << "TensorFlow Decision Forests instead "
+ << "(https://github.com/tensorflow/decision-forests).\n";
+ }
void Compute(OpKernelContext* const context) override {
// Get decision tree ensemble.
