1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
def if_enable_acl(if_true, if_false = []):
return select({
"@org_tensorflow//third_party/compute_library:build_with_acl": if_true,
"//conditions:default": if_false,
})
def acl_deps():
"""Returns the correct set of ACL library dependencies.
Shorthand for select() to pull in the correct set of ACL library deps
for aarch64 platform
Returns:
a select evaluating to a list of library dependencies, suitable for
inclusion in the deps attribute of rules.
"""
return select({
"@org_tensorflow//third_party/compute_library:build_with_acl": ["@compute_library//:arm_compute"],
"//conditions:default": [],
})
|
