diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-16 15:52:12 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-16 15:55:47 -0800 |
| commit | ccbd14b741e6efbe51769f0f1b9cb3719c42c23b (patch) | |
| tree | bcc79094e48982780b084cd19110d903186eea8f /tensorflow/contrib/tensor_forest | |
| parent | 287fe4f2404a7b69ffce89cc41ff3f049ca7a08b (diff) | |
Enable bfloat16 for CPU kernels
PiperOrigin-RevId: 182124532
Diffstat (limited to 'tensorflow/contrib/tensor_forest')
| -rw-r--r-- | tensorflow/contrib/tensor_forest/kernels/v4/BUILD | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/tensor_forest/kernels/v4/BUILD b/tensorflow/contrib/tensor_forest/kernels/v4/BUILD index b7876e1df6..794b76d858 100644 --- a/tensorflow/contrib/tensor_forest/kernels/v4/BUILD +++ b/tensorflow/contrib/tensor_forest/kernels/v4/BUILD @@ -302,6 +302,7 @@ cc_library( "//tensorflow/contrib/tensor_forest/proto:fertile_stats_proto_cc", ], [ + "//third_party/eigen3", "//tensorflow/contrib/decision_trees/proto:generic_tree_model_cc_headers_only", "//tensorflow/contrib/tensor_forest/proto:fertile_stats_proto_cc_headers_only", ], @@ -322,6 +323,7 @@ cc_library( srcs = ["params.cc"], hdrs = ["params.h"], deps = [ + "//third_party/eigen3", "//tensorflow/core:framework_headers_lib", ] + if_static( [ |