blob: 5d7e4a33f835b877eb2d7447fcecdb309dbe97d8 [file] [log] [blame]
Jeremy Johnson0ecfa372022-06-30 14:27:56 +01001{
2 "add": {
3 "tests": [
4 "add_1_qi16",
5 "add_1x32x32x8_qi8"
Jeremy Johnson88588622022-07-12 16:42:29 +01006 ],
7 "profile": [
8 "tosa-bi",
9 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +010010 ]
11 },
12 "average_pool_2d": {
13 "alternate_names": [
14 "avg_pool2d"
15 ],
16 "tests": [
17 "average_pool_2d_1x4x4x4_qi8_st11_padSAME_kern11",
18 "average_pool_2d_1x4x8x19_qi16_st21_padSAME_kern22",
19 "average_pool_2d_1x7x7x9_qi8_st22_padSAME_kern11",
20 "average_pool_2d_1x32x32x8_qu8_st12_padVALID_kern12",
21 "average_pool_2d_1x8x4x17_qu8_st21_padVALID_kern21"
Jeremy Johnson88588622022-07-12 16:42:29 +010022 ],
23 "profile": [
24 "tosa-bi",
25 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +010026 ]
27 },
28 "concatenation": {
29 "tests": [
30 "concatenation_concat_13x21x3_i32_axis_0",
31 "concatenation_concat_14x19_i32_axis_1",
32 "concatenation_concat_1x32x32x8_i32_axis_2",
33 "concatenation_concat_64_i32_axis_m1",
34 "concatenation_concatv2_13x21x3_i32_axis_1",
35 "concatenation_concatv2_1x32x32x8_i32_axis_m2",
36 "concatenation_concatv2_1x7x7x9_i32_axis_m3",
37 "concatenation_concatv2_64_i32_axis_0"
38 ],
39 "alternate_names": [
40 "concat",
41 "concatv2"
Jeremy Johnson88588622022-07-12 16:42:29 +010042 ],
43 "profile": [
44 "tosa-bi",
45 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +010046 ]
47 },
48 "conv_2d": {
49 "alternate_names": [
50 "conv2d"
51 ],
52 "tests": [
53 "conv_2d_1x1_1x4x8x19_qu8_st11_padVALID_dilat21",
54 "conv_2d_3x3_1x7x7x9_qi8_st12_padVALID_dilat11",
55 "conv_2d_5x5_1x32x32x8_qu8_st11_padSAME_dilat22",
56 "conv_2d_bias_1x1_1x32x32x8_qi8_st11_padVALID_dilat12",
57 "conv_2d_bias_5x5_1x7x7x9_qu8_st22_padSAME_dilat12",
58 "conv_2d_relu_1x1_1x32x32x8_qi8",
59 "conv_2d_relu_3x3_1x4x8x19_qu8",
60 "conv_2d_relu6_3x3_1x8x4x17_qi8",
61 "conv_2d_relu6_5x5_1x4x8x19_qu8",
62 "conv_2d_relu_n1_to_1_3x3_1x32x32x8_qu8",
63 "conv_2d_relu_n1_to_1_5x5_1x7x7x9_qi8",
64 "conv_2d_tanh_3x3_1x7x7x9_qi8",
65 "conv_2d_tanh_5x5_1x32x32x8_qu8"
Jeremy Johnson88588622022-07-12 16:42:29 +010066 ],
67 "profile": [
68 "tosa-bi",
69 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +010070 ]
71 },
72 "depthwise_conv_2d": {
73 "alternate_names": [
74 "depthwise_conv2d"
75 ],
76 "tests": [
77 "depthwise_conv_2d_1x1_1x32x32x8_qi8_st11_padSAME_dilat21",
78 "depthwise_conv_2d_3x3_1x7x7x9_qu8_st22_padVALID_dilat11",
79 "depthwise_conv_2d_5x5_1x32x32x8_qi8_st11_padSAME_dilat12",
80 "depthwise_conv_2d_bias_1x1_1x4x4x4_qu8_st11_padSAME_dilat11",
81 "depthwise_conv_2d_bias_3x3_1x32x32x8_qu8_st11_padVALID_dilat22"
Jeremy Johnson88588622022-07-12 16:42:29 +010082 ],
83 "profile": [
84 "tosa-bi",
85 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +010086 ]
87 },
Won Jeon78155c62023-06-10 00:20:04 +000088 "erf": {
89 "tests": [
90 "erf_13x21x3_qu8",
91 "erf_14x19_qi8",
92 "erf_1x8x4x17_qi16"
93 ],
94 "profile": [
95 "tosa-mi"
96 ]
97 },
Jeremy Johnson0ecfa372022-06-30 14:27:56 +010098 "fully_connected": {
99 "alternate_names": [
100 "matmul"
101 ],
102 "tests": [
103 "fully_connected_1x4x4x4_qi8",
104 "fully_connected_13x21x3_qu8",
105 "fully_connected_14x19_qi8"
Jeremy Johnson88588622022-07-12 16:42:29 +0100106 ],
107 "profile": [
108 "tosa-bi",
109 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100110 ]
111 },
112 "leaky_relu": {
113 "tests": [
114 "leaky_relu_1_qi16_0",
115 "leaky_relu_14x19_qi8_0",
116 "leaky_relu_13x21x3_qu8_0",
117 "leaky_relu_1x4x8x19_qi16_0"
Jeremy Johnson88588622022-07-12 16:42:29 +0100118 ],
119 "profile": [
120 "tosa-bi",
121 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100122 ]
123 },
124 "logistic": {
125 "alternate_names": [
126 "sigmoid"
127 ],
128 "tests": [
129 "logistic_13x21x3_qu8",
130 "logistic_1x8x4x17_qi8",
131 "logistic_64_qi16"
Jeremy Johnson88588622022-07-12 16:42:29 +0100132 ],
133 "profile": [
134 "tosa-bi",
135 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100136 ]
137 },
138 "max_pool_2d": {
139 "alternate_names": [
140 "max_pool2d"
141 ],
142 "tests": [
143 "max_pool_2d_1x4x4x4_qi8_st11_padSAME_kern11",
144 "max_pool_2d_1x4x4x4_qu8_st22_padVALID_kern22",
145 "max_pool_2d_1x4x8x19_qu8_st21_padVALID_kern22",
146 "max_pool_2d_1x8x4x17_qu8_st21_padVALID_kern21",
147 "max_pool_2d_1x32x32x8_qi8_st12_padVALID_kern22"
Jeremy Johnson88588622022-07-12 16:42:29 +0100148 ],
149 "profile": [
150 "tosa-bi",
151 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100152 ]
153 },
154 "minimum": {
155 "alternate_names": [
156 "min"
157 ],
158 "tests": [
159 "minimum_1x4x4x4_i32",
160 "minimum_64_i32"
Jeremy Johnson88588622022-07-12 16:42:29 +0100161 ],
162 "profile": [
163 "tosa-bi",
164 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100165 ]
166 },
167 "mul": {
168 "tests": [
169 "mul_13x21x3_qi16",
170 "mul_1x4x8x19_qu8"
Jeremy Johnson88588622022-07-12 16:42:29 +0100171 ],
172 "profile": [
173 "tosa-bi",
174 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100175 ]
176 },
177 "pack": {
178 "alternate_names": [
179 "stack"
180 ],
181 "tests": [
182 "pack_13x21x3_i32_axis0",
183 "pack_14x19_i32_axis1",
184 "pack_1x7x7x9_i32_axis2",
185 "pack_1x8x4x17_i32_axis4"
Jeremy Johnson88588622022-07-12 16:42:29 +0100186 ],
187 "profile": [
188 "tosa-bi",
189 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100190 ]
191 },
192 "reshape": {
193 "tests": [
194 "reshape_14x19_i32_rank1",
195 "reshape_13x21x3_i32_rank3",
196 "reshape_1x32x32x8_i32_rank2"
Jeremy Johnson88588622022-07-12 16:42:29 +0100197 ],
198 "profile": [
199 "tosa-bi",
200 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100201 ]
202 },
203 "resize_bilinear": {
Jeremy Johnsone81b2ba2023-03-13 12:21:22 +0000204 "alternate_names": [
205 "resize"
206 ],
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100207 "tests": [
Jeremy Johnsone81b2ba2023-03-13 12:21:22 +0000208 "resize_bilinear_1x32x32x8_qi16_bilinear_align0_half1_scale3",
209 "resize_bilinear_1x4x4x4_qi8_bilinear_align1_half0_scale1"
Jeremy Johnson88588622022-07-12 16:42:29 +0100210 ],
211 "profile": [
212 "tosa-bi",
213 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100214 ]
215 },
216 "resize_nearest_neighbor": {
217 "alternate_names": [
Jeremy Johnsone81b2ba2023-03-13 12:21:22 +0000218 "resize"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100219 ],
220 "tests": [
Jeremy Johnsone81b2ba2023-03-13 12:21:22 +0000221 "resize_nearest_neighbor_1x4x8x19_qu8_nearest_align0_half1_scale2",
222 "resize_nearest_neighbor_1x8x4x17_qi8_nearest_align0_half0_scale3"
Jeremy Johnson88588622022-07-12 16:42:29 +0100223 ],
224 "profile": [
225 "tosa-bi",
226 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100227 ]
228 },
229 "softmax": {
230 "tests": [
231 "softmax_1_qi8",
232 "softmax_14x19_qi16",
233 "softmax_1x32x32x8_qu8"
Jeremy Johnson88588622022-07-12 16:42:29 +0100234 ],
235 "profile": [
236 "tosa-bi",
237 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100238 ]
239 },
240 "split": {
241 "tests": [
Jeremy Johnsone81b2ba2023-03-13 12:21:22 +0000242 "split_3x1x1x7_i32_splitv_axis0",
Jeremy Johnson60136852023-02-09 12:16:14 +0000243 "split_13x21x3_i32_split3_axis1",
244 "split_1x4x8x19_i32_splitv_axis2",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100245 "split_64_i32_split2_axis0"
Jeremy Johnson88588622022-07-12 16:42:29 +0100246 ],
247 "profile": [
248 "tosa-bi",
249 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100250 ]
251 },
252 "strided_slice": {
253 "tests": [
254 "strided_slice_13x21x3_i32_perm0",
255 "strided_slice_14x19_i32_perm1",
256 "strided_slice_64_i32_perm3"
Jeremy Johnson88588622022-07-12 16:42:29 +0100257 ],
258 "profile": [
259 "tosa-bi",
260 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100261 ]
262 },
263 "tanh": {
264 "tests": [
265 "tanh_13x21x3_qu8",
266 "tanh_14x19_qi8",
267 "tanh_1x8x4x17_qi16"
Jeremy Johnson88588622022-07-12 16:42:29 +0100268 ],
269 "profile": [
Jeremy Johnson88588622022-07-12 16:42:29 +0100270 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100271 ]
272 }
Won Jeon78155c62023-06-10 00:20:04 +0000273}