From: Guozhong Zhuang <guozhong.zhuang@intel.com>
Date: Thu, 24 May 2018 20:30:18 +0000 (-0700)
Subject: fix an registration issue for MKL_ML op
X-Git-Tag: upstream/v1.9.0_rc1~53^2
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=6cb8af030cf7169e4c68361706b6af94545e18c3;p=platform%2Fupstream%2Ftensorflow.git

fix an registration issue for MKL_ML op
---

diff --git a/tensorflow/core/kernels/mkl_conv_grad_bias_ops.cc b/tensorflow/core/kernels/mkl_conv_grad_bias_ops.cc
index d23027a..a6698a1 100644
--- a/tensorflow/core/kernels/mkl_conv_grad_bias_ops.cc
+++ b/tensorflow/core/kernels/mkl_conv_grad_bias_ops.cc
@@ -18,6 +18,7 @@ limitations under the License.
 // bias.
 
 #ifdef INTEL_MKL
+#ifdef INTEL_MKL_ML
 
 #define USE_EIGEN_TENSOR
 #define EIGEN_USE_THREADS
@@ -262,4 +263,5 @@ class MklConv2DCustomBackpropBiasOp : public OpKernel {
 TF_CALL_float(REGISTER_CPU_KERNELS);
 #undef REGISTER_CPU_KERNELS
 } /* namespace tensorflow */
+#endif /* INTEL_MKL_ML */
 #endif /* INTEL_MKL */