diff options
Diffstat (limited to 'tensorflow/core/kernels/random_op.h')
-rw-r--r-- | tensorflow/core/kernels/random_op.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/random_op.h b/tensorflow/core/kernels/random_op.h index b52901c38e..97bcaf1a49 100644 --- a/tensorflow/core/kernels/random_op.h +++ b/tensorflow/core/kernels/random_op.h @@ -54,6 +54,18 @@ struct FillPhiloxRandom<GPUDevice, Distribution> { }; #endif // GOOGLE_CUDA +#if TENSORFLOW_USE_SYCL +typedef Eigen::SyclDevice SYCLDevice; +// Declares the partially SYCL-specialized functor struct. +template <class Distribution> +struct FillPhiloxRandom<SYCLDevice, Distribution> { + void operator()(OpKernelContext* ctx, const SYCLDevice& d, + random::PhiloxRandom gen, + typename Distribution::ResultElementType* data, int64 size, + Distribution dist); +}; +#endif // TENSORFLOW_USE_SYCL + } // namespace functor } // namespace tensorflow |