Default to TF threading for MKL-DNN

Author: Roman Dubtsov

.bazelrc

@@ -26,13 +26,15 @@ build --define framework_shared_object=true
 # environment variable "TF_MKL_ROOT" every time before build.
 build:mkl --define=build_with_mkl=true --define=enable_mkl=true
 build:mkl --define=tensorflow_mkldnn_contraction_kernel=0
+build:mkl --define mkl_dnn_threading=tf
 build:mkl -c opt
 
 # This config option is used to enable MKL-DNN open source library only,
 # without depending on MKL binary version.
 build:mkl_open_source_only --define=build_with_mkl_dnn_only=true
 build:mkl_open_source_only --define=build_with_mkl=true --define=enable_mkl=true
 build:mkl_open_source_only --define=tensorflow_mkldnn_contraction_kernel=0
+build:mkl_open_source_only --define mkl_dnn_threading=tf
 
 build:download_clang --crosstool_top=@local_config_download_clang//:toolchain
 build:download_clang --define=using_clang=true

tensorflow/workspace.bzl

@@ -118,11 +118,11 @@ def tf_workspace(path_prefix = "", tf_repo_name = ""):
     tf_http_archive(
         name = "mkl_dnn",
         build_file = clean_dep("//third_party/mkl_dnn:mkldnn.BUILD"),
-        sha256 = "bd406f4d9d004be270818449d010fe6cecd27af8cd465b5ec5e596905b45bb0d",
-        strip_prefix = "mkl-dnn-e842faaaf053c4e2926c3f857944c9837366c731",
+        sha256 = "ad1eff102f30e44d75c25abab118428f8c93d0b5bb28bb235d2df36acc7b54bf",
+        strip_prefix = "mkl-dnn-0154b64533a9d3f9f711ffadc967d6ed6fc59205",
         urls = [
-            "https://mirror.bazel.build/github.com/rsdubtso/mkl-dnn/archive/e842faaaf053c4e2926c3f857944c9837366c731.tar.gz",
-            "https://github.com/rsdubtso/mkl-dnn/archive/e842faaaf053c4e2926c3f857944c9837366c731.tar.gz",
+            "https://mirror.bazel.build/github.com/rsdubtso/mkl-dnn/archive/0154b64533a9d3f9f711ffadc967d6ed6fc59205.tar.gz",
+            "https://github.com/rsdubtso/mkl-dnn/archive/0154b64533a9d3f9f711ffadc967d6ed6fc59205.tar.gz",
         ],
     )