diff options
| author |  Andrew Harp <andrewharp@google.com> | 2017-03-01 17:59:22 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-01 18:08:24 -0800 |
| commit | 3e975ea978bac4d861bb09328b06f3c316212611 (patch) | |
| tree | 79bac044c9723df8443495eb962c2dd98a2ed421 /tensorflow/core/kernels/reverse_op.cc | |
| parent | 8043a27ed77f59bb68409070f2bfa01df0e04b89 (diff) | |
Merge changes from github.
Change: 148954491
Diffstat (limited to 'tensorflow/core/kernels/reverse_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/reverse_op.cc | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/reverse_op.cc b/tensorflow/core/kernels/reverse_op.cc index ec3b27ed6f..24b3ba31b8 100644 --- a/tensorflow/core/kernels/reverse_op.cc +++ b/tensorflow/core/kernels/reverse_op.cc @@ -33,6 +33,9 @@ namespace tensorflow { typedef Eigen::ThreadPoolDevice CPUDevice; typedef Eigen::GpuDevice GPUDevice; +#ifdef TENSORFLOW_USE_SYCL +typedef Eigen::SyclDevice SYCLDevice; +#endif // TENSORFLOW_USE_SYCL namespace { @@ -344,4 +347,36 @@ REGISTER_KERNEL_BUILDER(Name("ReverseV2") ReverseV2Op<CPUDevice, int32>); #endif // GOOGLE_CUDA +#ifdef TENSORFLOW_USE_SYCL +#define REGISTER_SYCL_KERNELS(T) \ + REGISTER_KERNEL_BUILDER(Name("Reverse") \ + .Device(DEVICE_SYCL) \ + .TypeConstraint<T>("T") \ + .HostMemory("dims"), \ + ReverseOp<SYCLDevice, T>) \ + REGISTER_KERNEL_BUILDER(Name("ReverseV2") \ + .Device(DEVICE_SYCL) \ + .TypeConstraint<T>("T") \ + .TypeConstraint<int32>("Tidx") \ + .HostMemory("axis"), \ + ReverseV2Op<SYCLDevice, T>) +TF_CALL_float(REGISTER_SYCL_KERNELS); + +REGISTER_KERNEL_BUILDER(Name("Reverse") + .Device(DEVICE_SYCL) + .TypeConstraint<int32>("T") + .HostMemory("tensor") + .HostMemory("dims") + .HostMemory("output"), + ReverseOp<CPUDevice, int32>); +REGISTER_KERNEL_BUILDER(Name("ReverseV2") + .Device(DEVICE_SYCL) + .TypeConstraint<int32>("T") + .TypeConstraint<int32>("Tidx") + .HostMemory("tensor") + .HostMemory("axis") + .HostMemory("output"), + ReverseV2Op<CPUDevice, int32>); +#endif // TENSORFLOW_USE_SYCL + } // namespace tensorflow |