diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD
index 6958d708c41..8d7c893a241 100644
--- a/tensorflow/core/kernels/BUILD
+++ b/tensorflow/core/kernels/BUILD
@@ -6074,6 +6074,22 @@ filegroup(
         "//tensorflow/core/data:stats_utils.h",
         "//tensorflow/core/data:unbounded_thread_pool.h",
         "//tensorflow/core/kernels/boosted_trees/quantiles:weighted_quantiles_hdrs",
+	"//tensorflow/core/framework:op_kernel.h",
+	"//tensorflow/core/framework:resource_mgr.h",
+	"//tensorflow/core/framework:tensor.h",
+	"//tensorflow/core/framework:tensor_shape.h",
+	"//tensorflow/core/framework:tensor_types.h",
+	"//tensorflow/core/framework:device_base.h",
+	"//tensorflow/core/kernels/boosted_trees:boosted_trees.pb.h",
+	"//tensorflow/core/kernels/boosted_trees:resources.h",
+	"//tensorflow/core/lib/core:errors.h",
+	"//tensorflow/core/lib/core:refcount.h",
+	"//tensorflow/core/lib/core:status.h",
+	"//tensorflow/core/lib/core:threadpool.h",
+	"//tensorflow/core/platform:mutex.h",
+	"//tensorflow/core/platform:protobuf.h",
+	"//tensorflow/core/platform:types.h",
+	"//tensorflow/core/util:work_sharder.h",
         "//tensorflow/core/kernels/data:batch_dataset_op.h",
         "//tensorflow/core/kernels/data:iterator_ops.h",
         "//tensorflow/core/kernels/data:finalize_dataset_op.h",
@@ -6411,6 +6427,8 @@ filegroup(
         "xent_op.cc",
     ] + [
         "//tensorflow/core/kernels/boosted_trees:quantile_ops.cc",
+	"//tensorflow/core/kernels/boosted_trees:prediction_ops.cc",
+	"//tensorflow/core/kernels/boosted_trees:resource_ops.cc",
         "//tensorflow/core/data:captured_function.cc",
         "//tensorflow/core/data:dataset_utils.cc",
         "//tensorflow/core/data:name_utils.cc",
diff --git a/tensorflow/lite/BUILD b/tensorflow/lite/BUILD
index 72a46e6675f..7b8e866d1af 100644
--- a/tensorflow/lite/BUILD
+++ b/tensorflow/lite/BUILD
@@ -1050,7 +1050,8 @@ tflite_cc_shared_object(
         ":framework",
         ":tflite_exported_symbols.lds",
         ":tflite_version_script.lds",
-        "//tensorflow/lite/kernels:builtin_ops_all_linked",
+	"//tensorflow/lite/kernels:builtin_ops",
+	"//tensorflow/lite/delegates/flex:delegate",
     ],
 )
 
diff --git a/tensorflow/lite/c/BUILD b/tensorflow/lite/c/BUILD
index 9fc28c9d3da..8a4d19193bd 100644
--- a/tensorflow/lite/c/BUILD
+++ b/tensorflow/lite/c/BUILD
@@ -38,6 +38,7 @@ tflite_cc_shared_object(
         ":c_api_experimental",
         ":exported_symbols.lds",
         ":version_script.lds",
+	"//tensorflow/lite/delegates/flex:delegate",
     ],
 )
 
diff --git a/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc b/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
index c42bd76a044..0f5c36e3619 100644
--- a/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
+++ b/tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc
@@ -105,6 +105,8 @@ const std::set<std::string>& GetFlexAllowlist() {
           "BoostedTreesQuantileStreamResourceFlush",
           "BoostedTreesQuantileStreamResourceGetBucketBoundaries",
           "BoostedTreesQuantileStreamResourceHandleOp",
+	  "BoostedTreesEnsembleResourceHandleOp",
+	  "BoostedTreesPredict",
           "BroadcastArgs",
           "BroadcastGradientArgs",
           "BroadcastTo",
diff --git a/tensorflow/lite/g3doc/guide/op_select_allowlist.md b/tensorflow/lite/g3doc/guide/op_select_allowlist.md
index 52f6e9c2822..81926acba61 100644
--- a/tensorflow/lite/g3doc/guide/op_select_allowlist.md
+++ b/tensorflow/lite/g3doc/guide/op_select_allowlist.md
@@ -84,6 +84,8 @@ supported by TensorFlow Lite runtime with the Select TensorFlow Ops feature.
 *   `raw_ops.BoostedTreesQuantileStreamResourceFlush`
 *   `raw_ops.BoostedTreesQuantileStreamResourceGetBucketBoundaries`
 *   `raw_ops.BoostedTreesQuantileStreamResourceHandleOp`
+*   `raw_ops.BoostedTreesEnsembleResourceHandleOp`
+*   `raw_ops.BoostedTreesPredict`
 *   `raw_ops.BroadcastArgs`
 *   `raw_ops.BroadcastGradientArgs`
 *   `raw_ops.BroadcastTo`