mirror of
https://github.com/zebrajr/tensorflow.git
synced 2025-12-06 12:20:11 +01:00
fix broken bazel macros, to fix TF serving build.
This commit is contained in:
parent
6ce39236b6
commit
9be40198b2
8
third_party/mkl/build_defs.bzl
vendored
8
third_party/mkl/build_defs.bzl
vendored
|
|
@ -33,8 +33,8 @@ def if_mkl(if_true, if_false = []):
|
|||
"""
|
||||
return select({
|
||||
"@org_tensorflow//third_party/mkl:build_with_mkl_aarch64": if_true,
|
||||
"//tensorflow:linux_x86_64": if_true,
|
||||
"//tensorflow:windows": if_true,
|
||||
"@org_tensorflow//tensorflow:linux_x86_64": if_true,
|
||||
"@org_tensorflow//tensorflow:windows": if_true,
|
||||
"//conditions:default": if_false,
|
||||
})
|
||||
|
||||
|
|
@ -102,8 +102,8 @@ def mkl_deps():
|
|||
"""
|
||||
return select({
|
||||
"@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"],
|
||||
"//tensorflow:windows": ["@mkl_dnn_v1//:mkl_dnn"],
|
||||
"@org_tensorflow//tensorflow:linux_x86_64": ["@mkl_dnn_v1//:mkl_dnn"],
|
||||
"@org_tensorflow//tensorflow:windows": ["@mkl_dnn_v1//:mkl_dnn"],
|
||||
"//conditions:default": [],
|
||||
})
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user