diff options
Diffstat (limited to 'unsupported/Eigen/CXX11/Tensor')
-rw-r--r-- | unsupported/Eigen/CXX11/Tensor | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/Tensor b/unsupported/Eigen/CXX11/Tensor index 14a2d0f0d..420afe650 100644 --- a/unsupported/Eigen/CXX11/Tensor +++ b/unsupported/Eigen/CXX11/Tensor @@ -75,6 +75,10 @@ typedef unsigned __int64 uint64_t; #include "libxsmm.h" #endif +#if defined(EIGEN_USE_MKLDNN) +#include "mkldnn.h" +#endif + #ifdef EIGEN_USE_THREADS #include "ThreadPool" #endif @@ -121,6 +125,7 @@ typedef unsigned __int64 uint64_t; #include "src/Tensor/TensorArgMax.h" #include "src/Tensor/TensorConcatenation.h" #include "src/Tensor/TensorContractionMapper.h" +#include "src/Tensor/TensorContractionMkldnn.h" #include "src/Tensor/TensorContractionBlocking.h" #include "src/Tensor/TensorContraction.h" #include "src/Tensor/TensorContractionThreadPool.h" |