diff options
Diffstat (limited to 'tensorflow/core/kernels/mkl_conv_grad_bias_ops.cc')
-rw-r--r-- | tensorflow/core/kernels/mkl_conv_grad_bias_ops.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/mkl_conv_grad_bias_ops.cc b/tensorflow/core/kernels/mkl_conv_grad_bias_ops.cc index 8a1006a8e9..d4364d31e4 100644 --- a/tensorflow/core/kernels/mkl_conv_grad_bias_ops.cc +++ b/tensorflow/core/kernels/mkl_conv_grad_bias_ops.cc @@ -38,9 +38,9 @@ limitations under the License. #include "tensorflow/core/util/use_cudnn.h" #include "tensorflow/core/util/work_sharder.h" +#include "tensorflow/core/util/mkl_util.h" #include "third_party/mkl/include/mkl_dnn.h" #include "third_party/mkl/include/mkl_dnn_types.h" -#include "tensorflow/core/util/mkl_util.h" namespace tensorflow { @@ -252,7 +252,7 @@ class MklConv2DCustomBackpropBiasOp : public OpKernel { }; #define REGISTER_CPU_KERNELS(T) \ - REGISTER_KERNEL_BUILDER(Name("_MklConv2DWithBiasBackpropBias") \ + REGISTER_KERNEL_BUILDER(Name("_MklConv2DWithBiasBackpropBias") \ .Device(DEVICE_CPU) \ .TypeConstraint<T>("T") \ .Label(mkl_op_registry::kMklOpLabel), \ |