diff options
| author |  Sourabh Bajaj <sourabhbajaj@google.com> | 2017-11-30 16:37:11 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-30 16:41:01 -0800 |
| commit | b2db981a6731e978453862a73dab892bc674db68 (patch) | |
| tree | c11a7c4038e2595268113c2859c1d0d3072ede4f /tensorflow/core/kernels/batch_matmul_op_complex.cc | |
| parent | 0438ac79bdb503ed267bec2146e7136ac8e99ff9 (diff) | |
Merge changes from github.
PiperOrigin-RevId: 177526301
Diffstat (limited to 'tensorflow/core/kernels/batch_matmul_op_complex.cc')
| -rw-r--r-- | tensorflow/core/kernels/batch_matmul_op_complex.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/batch_matmul_op_complex.cc b/tensorflow/core/kernels/batch_matmul_op_complex.cc index a58ec02726..96216764fd 100644 --- a/tensorflow/core/kernels/batch_matmul_op_complex.cc +++ b/tensorflow/core/kernels/batch_matmul_op_complex.cc @@ -17,8 +17,10 @@ limitations under the License. namespace tensorflow { +#if !defined(INTEL_MKL) TF_CALL_complex64(REGISTER_BATCH_MATMUL_CPU); TF_CALL_complex128(REGISTER_BATCH_MATMUL_CPU); +#endif #if GOOGLE_CUDA TF_CALL_complex64(REGISTER_BATCH_MATMUL_GPU); |