diff options
Diffstat (limited to 'tensorflow/core/kernels/BUILD')
| -rw-r--r-- | tensorflow/core/kernels/BUILD | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index 2503b441f7..fb7d63767d 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -39,6 +39,10 @@ load( "tf_proto_library", "tf_kernel_tests_linkstatic", ) +load( + "//third_party/mkl:build_defs.bzl", + "if_mkl", +) config_setting( # Add "--define tensorflow_xsmm=1" to your build command to use libxsmm for @@ -4294,6 +4298,17 @@ tf_cc_test( ], ) +if_mkl( + tf_kernel_library( + name = "mkl_ops", + prefix = "mkl_matmul", + deps = [ + ":math", + "//third_party/mkl:intel_binary_blob", + ], + ), +) + # ----------------------------------------------------------------------------- # Google-internal targets. These must be at the end for syncrepo. |