diff options
Diffstat (limited to 'tensorflow/core/kernels/mkl_tfconv_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/mkl_tfconv_op.cc | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/core/kernels/mkl_tfconv_op.cc b/tensorflow/core/kernels/mkl_tfconv_op.cc index 51f90b3f90..c31ef5c255 100644 --- a/tensorflow/core/kernels/mkl_tfconv_op.cc +++ b/tensorflow/core/kernels/mkl_tfconv_op.cc @@ -105,11 +105,11 @@ class MklToTfOp : public OpKernel { // Register kernel /////////////////////////////////////////////////////////// -#define REGISTER_CPU(T) \ - REGISTER_KERNEL_BUILDER(Name("MklToTf") \ - .Device(DEVICE_CPU) \ - .TypeConstraint<T>("T") \ - .Label(mkl_layer_registry::kMklLayerLabel), \ +#define REGISTER_CPU(T) \ + REGISTER_KERNEL_BUILDER(Name("MklToTf") \ + .Device(DEVICE_CPU) \ + .TypeConstraint<T>("T") \ + .Label(mkl_op_registry::kMklOpLabel), \ MklToTfOp<CPUDevice, T>); TF_CALL_float(REGISTER_CPU); |