diff options
Diffstat (limited to 'tensorflow/core/kernels/concat_lib.h')
-rw-r--r-- | tensorflow/core/kernels/concat_lib.h | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/concat_lib.h b/tensorflow/core/kernels/concat_lib.h index cef873f804..14e6e1bc32 100644 --- a/tensorflow/core/kernels/concat_lib.h +++ b/tensorflow/core/kernels/concat_lib.h @@ -38,6 +38,14 @@ void ConcatGPU( Tensor* output, typename TTypes<T, 2>::Tensor* output_flat); #endif // GOOGLE_CUDA + +#ifdef TENSORFLOW_USE_SYCL +template <typename T> +void ConcatSYCL(const Eigen::SyclDevice& d, + const std::vector< + std::unique_ptr<typename TTypes<T, 2>::ConstMatrix>>& inputs, + typename TTypes<T, 2>::Matrix* output); +#endif // TENSORFLOW_USE_SYCL } // namespace tensorflow #endif // TENSORFLOW_KERNELS_CONCAT_LIB_H_ |