OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix broken bazel macros, to fix TF serving build.

Browse Source
This commit is contained in:
Abhijit Karmarkar 2021-03-25 14:50:30 -07:00
parent 6ce39236b6
commit 9be40198b2
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
third_party/mkl/build_defs.bzl vendored
View File

@ -33,8 +33,8 @@ def if_mkl(if_true, if_false = []):
""" """
return select({ return select({
"@org_tensorflow//third_party/mkl:build_with_mkl_aarch64": if_true, "@org_tensorflow//third_party/mkl:build_with_mkl_aarch64": if_true,
"//tensorflow:linux_x86_64": if_true, "@org_tensorflow//tensorflow:linux_x86_64": if_true,
"//tensorflow:windows": if_true, "@org_tensorflow//tensorflow:windows": if_true,
"//conditions:default": if_false, "//conditions:default": if_false,
}) })
@ -102,8 +102,8 @@ def mkl_deps():
""" """
return select({ return select({
"@org_tensorflow//third_party/mkl:build_with_mkl_aarch64": ["@mkl_dnn_acl_compatible//:mkl_dnn_acl"], "@org_tensorflow//third_party/mkl:build_with_mkl_aarch64": ["@mkl_dnn_acl_compatible//:mkl_dnn_acl"],
"//tensorflow:linux_x86_64": ["@mkl_dnn_v1//:mkl_dnn"], "@org_tensorflow//tensorflow:linux_x86_64": ["@mkl_dnn_v1//:mkl_dnn"],
"//tensorflow:windows": ["@mkl_dnn_v1//:mkl_dnn"], "@org_tensorflow//tensorflow:windows": ["@mkl_dnn_v1//:mkl_dnn"],
"//conditions:default": [], "//conditions:default": [],
}) })