diff options
Diffstat (limited to 'tensorflow/core/kernels/transpose_functor.h')
| -rw-r--r-- | tensorflow/core/kernels/transpose_functor.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/transpose_functor.h b/tensorflow/core/kernels/transpose_functor.h index 124cf14dd2..f1ab770eeb 100644 --- a/tensorflow/core/kernels/transpose_functor.h +++ b/tensorflow/core/kernels/transpose_functor.h @@ -132,6 +132,13 @@ template <typename Device, typename T, int NDIMS> void TransposeUsingEigen(const Device& d, const Tensor& in, const gtl::ArraySlice<int32> perm, Tensor* out); + +#ifdef TENSORFLOW_USE_SYCL +// For SYCL lets always go through Eigen +template <typename Device, typename T> +void TransposeSYCL(const Device& d, const Tensor& in, + const gtl::ArraySlice<int32> perm, Tensor* out); +#endif // TENSORFLOW_USE_SYCL } // namespace internal template <typename Device, typename T> |