diff options
Diffstat (limited to 'tensorflow/core/framework/op_kernel.cc')
| -rw-r--r-- | tensorflow/core/framework/op_kernel.cc | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/core/framework/op_kernel.cc b/tensorflow/core/framework/op_kernel.cc index a7a93cb69c..50520bb3fd 100644 --- a/tensorflow/core/framework/op_kernel.cc +++ b/tensorflow/core/framework/op_kernel.cc @@ -949,6 +949,13 @@ const Eigen::GpuDevice& OpKernelContext::eigen_device() const { return eigen_gpu_device(); } +#ifdef TENSORFLOW_USE_SYCL +template <> +const Eigen::SyclDevice& OpKernelContext::eigen_device() const { + return eigen_sycl_device(); +} +#endif + void OpKernelConstruction::CtxFailure(Status s) { VLOG(1) << s; SetStatus(s); |