experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

enable mkl in eigen for qr_op (#11116)

Browse Source
This commit is contained in:
Ziming Dong 2017-08-11 01:43:06 +08:00 committed by Rasmus Munk Larsen
parent ed7d08eaf9
commit 17e82cc274
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/qr_op_impl.h
View File

@ -20,6 +20,10 @@ limitations under the License.
// improve compilation times.
#include <algorithm>
#ifdef INTEL_MKL
#define EIGEN_USE_MKL_ALL
#endif // INTEL_MKL
#include "third_party/eigen3/Eigen/QR"
#include "tensorflow/core/framework/kernel_def_builder.h"
#include "tensorflow/core/framework/op_kernel.h"