diff options
| author |  AG Ramesh <ag.ramesh@intel.com> | 2018-06-08 15:06:47 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-08 15:09:15 -0700 |
| commit | db717a72c20ab37974ec9076c8e406345c8776be (patch) | |
| tree | c66d715479275dbf3cb51f553b44f202777b7643 /tensorflow/core/kernels/mkl_relu_op.cc | |
| parent | bc65583b2b4e3f48b6a724832ef96ab176666d33 (diff) | |
[INTEL MKL] Enable compilation of TF without MKL ML dependency
Closes #19808.
PiperOrigin-RevId: 199857219
Diffstat (limited to 'tensorflow/core/kernels/mkl_relu_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/mkl_relu_op.cc | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/mkl_relu_op.cc b/tensorflow/core/kernels/mkl_relu_op.cc index 1ed43834dd..78abbdb730 100644 --- a/tensorflow/core/kernels/mkl_relu_op.cc +++ b/tensorflow/core/kernels/mkl_relu_op.cc @@ -23,9 +23,6 @@ limitations under the License. #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/lib/core/errors.h" -#include "mkl_dnn.h" -#include "mkl_dnn_types.h" -#include "tensorflow/core/util/mkl_util.h" #ifndef INTEL_MKL_ML #include "mkldnn.hpp" @@ -38,7 +35,11 @@ using mkldnn::prop_kind; using mkldnn::relu_backward; using mkldnn::relu_forward; using mkldnn::stream; +#else +#include "mkl_dnn.h" +#include "mkl_dnn_types.h" #endif +#include "tensorflow/core/util/mkl_util.h" namespace tensorflow { |