From 386558b81f295277a1e79c540f3cb8d9c748fc25 Mon Sep 17 00:00:00 2001 From: "Wen-Heng (Jack) Chung" Date: Wed, 13 Mar 2019 19:11:56 +0000 Subject: [PATCH] Fix build errors --- tensorflow/core/kernels/concat_lib_gpu.h | 2 +- tensorflow/core/kernels/split_lib_gpu.h | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/core/kernels/concat_lib_gpu.h b/tensorflow/core/kernels/concat_lib_gpu.h index f8898e6537b..3fcecd754fe 100644 --- a/tensorflow/core/kernels/concat_lib_gpu.h +++ b/tensorflow/core/kernels/concat_lib_gpu.h @@ -25,7 +25,7 @@ limitations under the License. #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" #include "tensorflow/core/framework/register_types.h" #include "tensorflow/core/kernels/concat_lib.h" -#include "tensorflow/core/kernels/cuda_device_array_gpu.h" +#include "tensorflow/core/kernels/gpu_device_array_gpu.h" namespace tensorflow { diff --git a/tensorflow/core/kernels/split_lib_gpu.h b/tensorflow/core/kernels/split_lib_gpu.h index c667068e34b..ff76d072319 100644 --- a/tensorflow/core/kernels/split_lib_gpu.h +++ b/tensorflow/core/kernels/split_lib_gpu.h @@ -24,7 +24,7 @@ limitations under the License. #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" #include "tensorflow/core/framework/register_types.h" -#include "tensorflow/core/kernels/cuda_device_array_gpu.h" +#include "tensorflow/core/kernels/gpu_device_array_gpu.h" #include "tensorflow/core/kernels/split_lib.h" namespace tensorflow {