diff options
Diffstat (limited to 'tensorflow/core/kernels/mkl_matmul_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/mkl_matmul_op.cc | 16 |
1 files changed, 7 insertions, 9 deletions
diff --git a/tensorflow/core/kernels/mkl_matmul_op.cc b/tensorflow/core/kernels/mkl_matmul_op.cc index 3ba28c13ed..e43b75e250 100644 --- a/tensorflow/core/kernels/mkl_matmul_op.cc +++ b/tensorflow/core/kernels/mkl_matmul_op.cc @@ -199,15 +199,13 @@ class MklMatMulOp : public OpKernel { } }; -#define REGISTER_CPU(T) \ - REGISTER_KERNEL_BUILDER( \ - Name("MatMul").Device(DEVICE_CPU).TypeConstraint<T>("T"), \ - MklMatMulOp<CPUDevice, T, false /* cublas, ignored for CPU */>); \ - REGISTER_KERNEL_BUILDER( \ - Name("MatMul").Device(DEVICE_CPU).TypeConstraint<T>("T").Label("MKL"), \ - MklMatMulOp<CPUDevice, T, false /* cublas, ignored for CPU */>) - -// TODO:Consider template specialization when adding/removing additional types +#define REGISTER_CPU(T) \ + REGISTER_KERNEL_BUILDER( \ + Name("MatMul").Device(DEVICE_CPU).TypeConstraint<T>("T"), \ + MklMatMulOp<CPUDevice, T, false /* cublas, ignored for CPU */>); + +// TODO(inteltf) Consider template specialization when adding/removing +// additional types TF_CALL_float(REGISTER_CPU); TF_CALL_double(REGISTER_CPU); TF_CALL_complex64(REGISTER_CPU); |