Internal change

PiperOrigin-RevId: 258822180
This commit is contained in:
A. Unique TensorFlower 2019-07-18 12:29:03 -07:00 committed by TensorFlower Gardener
parent fe44fdddbc
commit c68ed58053

View File

@ -106,12 +106,12 @@ def mkl_deps():
inclusion in the deps attribute of rules. inclusion in the deps attribute of rules.
""" """
return select({ return select({
str(Label("//third_party/mkl_dnn:build_with_mkl_dnn_only")): ["@mkl_dnn"], str(Label("//third_party/mkl_dnn:build_with_mkl_dnn_only")): ["//third_party/intel_mkl_dnn"],
str(Label("//third_party/mkl_dnn:build_with_mkl_dnn_v1_only")): ["@mkl_dnn_v1//:mkl_dnn"], str(Label("//third_party/mkl_dnn:build_with_mkl_dnn_v1_only")): ["@mkl_dnn_v1//:mkl_dnn"],
str(Label("//third_party/mkl:build_with_mkl_ml_only")): ["//third_party/mkl:intel_binary_blob"], str(Label("//third_party/mkl:build_with_mkl_ml_only")): ["//third_party/mkl:intel_binary_blob"],
str(Label("//third_party/mkl:build_with_mkl")): [ str(Label("//third_party/mkl:build_with_mkl")): [
"//third_party/mkl:intel_binary_blob", "//third_party/mkl:intel_binary_blob",
"@mkl_dnn", "//third_party/intel_mkl_dnn",
], ],
"//conditions:default": [], "//conditions:default": [],
}) })