diff options
author | 2016-08-03 11:06:18 -0800 | |
---|---|---|
committer | 2016-08-03 12:18:00 -0700 | |
commit | c5b3ea14c0fc93032a8d31afa4546d7add1ed25e (patch) | |
tree | 5a07bd060d0b2cd85239b85d2551dd5cf9fc682e /tensorflow/contrib/quantization | |
parent | ec73b4e819e6dde705dba1e644648e01421ac523 (diff) |
Ensure custom GPU kernels are properly registered.
This change ensures that GOOGLE_CUDA=1 when compiling .cc files with --config=cuda enabled (also includes other important copts like -fno-exceptions.)
Change: 129242223
Diffstat (limited to 'tensorflow/contrib/quantization')
-rw-r--r-- | tensorflow/contrib/quantization/kernels/quantized_pooling_ops.cc | 5 |
1 files changed, 0 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; |