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