aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/random_op.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/kernels/random_op.h')
-rw-r--r--tensorflow/core/kernels/random_op.h12
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