diff options
-rw-r--r-- | tensorflow/contrib/tensor_forest/BUILD | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/contrib/tensor_forest/BUILD b/tensorflow/contrib/tensor_forest/BUILD index b8e8dbe127..40d2318311 100644 --- a/tensorflow/contrib/tensor_forest/BUILD +++ b/tensorflow/contrib/tensor_forest/BUILD @@ -35,6 +35,7 @@ filegroup( ], exclude = [ "kernels/*_test.cc", + "kernels/tree_utils.cc", ], ), ) @@ -45,6 +46,10 @@ filegroup( [ "kernels/*.h", ], + exclude = [ + "kernels/data_spec.h", + "kernels/tree_utils.h", + ], ), ) @@ -53,6 +58,7 @@ cc_library( srcs = [":custom_op_sources"], hdrs = [":custom_op_headers"], deps = [ + ":tree_utils", "//tensorflow/core:framework_headers_lib", "//third_party/eigen3", "@protobuf//:protobuf_headers", |