diff options
| -rw-r--r-- | tensorflow/contrib/quantization/kernels/quantized_pooling_ops.cc | 5 | ||||
| -rw-r--r-- | tensorflow/tensorflow.bzl | 1 | ||||
| -rw-r--r-- | tensorflow/tf_exported_symbols.lds | 1 | ||||
| -rw-r--r-- | tensorflow/tf_version_script.lds | 1 |
4 files changed, 3 insertions, 5 deletions
diff --git a/tensorflow/contrib/quantization/kernels/quantized_pooling_ops.cc b/tensorflow/contrib/quantization/kernels/quantized_pooling_ops.cc index c078de7ab1..33a12c4746 100644 --- a/tensorflow/contrib/quantization/kernels/quantized_pooling_ops.cc +++ b/tensorflow/contrib/quantization/kernels/quantized_pooling_ops.cc @@ -29,11 +29,6 @@ limitations under the License. #include "tensorflow/core/util/padding.h" #include "tensorflow/core/util/tensor_format.h" -#if GOOGLE_CUDA -#include "tensorflow/core/kernels/maxpooling_op_gpu.h" -#include "tensorflow/core/kernels/pooling_ops_common_gpu.h" -#endif // GOOGLE_CUDA - namespace tensorflow { typedef Eigen::ThreadPoolDevice CPUDevice; diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 8c92b0bf78..194309b134 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -634,6 +634,7 @@ def tf_custom_op_library(name, srcs=[], gpu_srcs=[], deps=[]): srcs=srcs, deps=deps + if_cuda(cuda_deps), data=[name + "_check_deps"], + copts=tf_copts(), linkshared=1, linkopts = select({ "//conditions:default": [ diff --git a/tensorflow/tf_exported_symbols.lds b/tensorflow/tf_exported_symbols.lds index e02e51eae0..e7749ab0f7 100644 --- a/tensorflow/tf_exported_symbols.lds +++ b/tensorflow/tf_exported_symbols.lds @@ -1 +1,2 @@ *tensorflow* +*perftools*gputools* diff --git a/tensorflow/tf_version_script.lds b/tensorflow/tf_version_script.lds index 61ffb8d29a..8c8c8be5a9 100644 --- a/tensorflow/tf_version_script.lds +++ b/tensorflow/tf_version_script.lds @@ -1,6 +1,7 @@ tensorflow { global: *tensorflow*; + *perftools*gputools*; local: *; }; |