diff options
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r-- | tensorflow/tensorflow.bzl | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 2f19147dbb..6d6e8941c5 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -31,12 +31,10 @@ load( "//third_party/mkl_dnn:build_defs.bzl", "if_mkl_open_source_only", ) - load( "//third_party/ngraph:build_defs.bzl", "if_ngraph", ) - def register_extension_info(**kwargs): pass @@ -398,7 +396,7 @@ def tf_cc_binary( srcs = srcs + tf_binary_additional_srcs(), deps = deps + tf_binary_dynamic_kernel_deps(kernels) + if_mkl_ml( [ - "//third_party/intel_mkl_ml", + "//third_party/mkl:intel_binary_blob", ], ), data = data + tf_binary_dynamic_kernel_dsos(kernels), @@ -736,7 +734,7 @@ def tf_cc_test( }) + linkopts + _rpath_linkopts(name), deps = deps + tf_binary_dynamic_kernel_deps(kernels) + if_mkl_ml( [ - "//third_party/intel_mkl_ml", + "//third_party/mkl:intel_binary_blob", ], ), data = data + tf_binary_dynamic_kernel_dsos(kernels), |