diff options
Diffstat (limited to 'tensorflow/core/kernels/relu_op.h')
| -rw-r--r-- | tensorflow/core/kernels/relu_op.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/relu_op.h b/tensorflow/core/kernels/relu_op.h index 365c6201a5..e2e0bd48dd 100644 --- a/tensorflow/core/kernels/relu_op.h +++ b/tensorflow/core/kernels/relu_op.h @@ -175,6 +175,10 @@ void EluGradOp<Device, T>::OperateNoTemplate(OpKernelContext* context, } // namespace tensorflow +#ifdef TENSORFLOW_USE_SYCL +#undef EIGEN_USE_SYCL +#endif // TENSORFLOW_USE_SYCL + #undef EIGEN_USE_THREADS #endif // TENSORFLOW_KERNELS_RELU_OP_H_ |