forked from XiaoMi/mobile-ai-bench
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathaibench.bzl
49 lines (41 loc) · 1009 Bytes
/
aibench.bzl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# -*- Python -*-
def if_android(a):
return select({
"//aibench:android": a,
"//conditions:default": [],
})
def if_not_android(a):
return select({
"//aibench:android": [],
"//conditions:default": a,
})
def if_android_armv7(a):
return select({
"//aibench:android_armv7": a,
"//conditions:default": [],
})
def if_android_arm64(a):
return select({
"//aibench:android_arm64": a,
"//conditions:default": [],
})
def if_mace_enabled(a):
return select({
"//aibench:mace_enabled": a,
"//conditions:default": [],
})
def if_ncnn_enabled(a):
return select({
"//aibench:ncnn_enabled": a,
"//conditions:default": [],
})
def if_snpe_enabled(a):
return select({
"//aibench:snpe_enabled": a,
"//conditions:default": [],
})
def if_tflite_enabled(a):
return select({
"//aibench:tflite_enabled": a,
"//conditions:default": [],
})