diff options
Diffstat (limited to 'tensorflow/core/common_runtime/sycl/sycl_device_context.cc')
| -rw-r--r-- | tensorflow/core/common_runtime/sycl/sycl_device_context.cc | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/tensorflow/core/common_runtime/sycl/sycl_device_context.cc b/tensorflow/core/common_runtime/sycl/sycl_device_context.cc index 9dd289bebd..b487d24c20 100644 --- a/tensorflow/core/common_runtime/sycl/sycl_device_context.cc +++ b/tensorflow/core/common_runtime/sycl/sycl_device_context.cc @@ -16,13 +16,11 @@ limitations under the License. #if TENSORFLOW_USE_SYCL #define EIGEN_USE_SYCL +#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" #include "tensorflow/core/common_runtime/sycl/sycl_device_context.h" #include "tensorflow/core/common_runtime/dma_helper.h" -#define EIGEN_USE_SYCL -#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" - namespace tensorflow { void SYCLDeviceContext::CopyCPUTensorToDevice(const Tensor *cpu_tensor, @@ -108,7 +106,6 @@ void SYCLDeviceContext::CopyDeviceTensorToCPU(const Tensor *device_tensor, StatusCallback done) { const int64 total_bytes = device_tensor->TotalBytes(); if (total_bytes > 0) { - device->eigen_sycl_device()->deallocate_all(); const void* src_ptr = DMAHelper::base(device_tensor); void* dst_ptr = DMAHelper::base(cpu_tensor); switch (device_tensor->dtype()) { |