diff options
author | Allen Lavoie <allenlavoie@gmail.com> | 2018-04-10 12:41:29 -0700 |
---|---|---|
committer | drpngx <drpngx@users.noreply.github.com> | 2018-04-10 12:41:29 -0700 |
commit | fe3f9dddb39171dd7cd9fbb9e044a40e08072c50 (patch) | |
tree | eceb77435174cc522c38f8864c31158604646f61 /tensorflow/BUILD | |
parent | 049dfd5e070cfa84c82eea71c6c746a70cba4a3f (diff) |
Make custom_graph_optimizer_registry header-only (#18387)
Adds it as a dependency to libtensorflow_framework.so so its symbols are
available to shared objects which want to register optimizers. No other rules
include it, so shared objects won't accidentally get their own version of the
registry.
Diffstat (limited to 'tensorflow/BUILD')
-rw-r--r-- | tensorflow/BUILD | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/BUILD b/tensorflow/BUILD index cfafffdd13..f2ad16fa04 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -450,11 +450,12 @@ tf_cc_shared_object( linkstatic = 1, visibility = ["//visibility:public"], deps = [ + "//tensorflow/core:core_cpu_impl", "//tensorflow/core:framework_internal_impl", + "//tensorflow/core:gpu_runtime_impl", + "//tensorflow/core/grappler/optimizers:custom_graph_optimizer_registry_impl", "//tensorflow/core:lib_internal_impl", - "//tensorflow/core:core_cpu_impl", "//tensorflow/stream_executor:stream_executor_impl", - "//tensorflow/core:gpu_runtime_impl", ] + tf_additional_binary_deps(), ) |