blob: 24a23b514f9a0d775d2addf42dac1150bca8a88d [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 },
88 "fully_connected": {
89 "alternate_names": [
90 "matmul"
91 ],
92 "tests": [
93 "fully_connected_1x4x4x4_qi8",
94 "fully_connected_13x21x3_qu8",
95 "fully_connected_14x19_qi8"
Jeremy Johnson88588622022-07-12 16:42:29 +010096 ],
97 "profile": [
98 "tosa-bi",
99 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100100 ]
101 },
102 "leaky_relu": {
103 "tests": [
104 "leaky_relu_1_qi16_0",
105 "leaky_relu_14x19_qi8_0",
106 "leaky_relu_13x21x3_qu8_0",
107 "leaky_relu_1x4x8x19_qi16_0"
Jeremy Johnson88588622022-07-12 16:42:29 +0100108 ],
109 "profile": [
110 "tosa-bi",
111 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100112 ]
113 },
114 "logistic": {
115 "alternate_names": [
116 "sigmoid"
117 ],
118 "tests": [
119 "logistic_13x21x3_qu8",
120 "logistic_1x8x4x17_qi8",
121 "logistic_64_qi16"
Jeremy Johnson88588622022-07-12 16:42:29 +0100122 ],
123 "profile": [
124 "tosa-bi",
125 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100126 ]
127 },
128 "max_pool_2d": {
129 "alternate_names": [
130 "max_pool2d"
131 ],
132 "tests": [
133 "max_pool_2d_1x4x4x4_qi8_st11_padSAME_kern11",
134 "max_pool_2d_1x4x4x4_qu8_st22_padVALID_kern22",
135 "max_pool_2d_1x4x8x19_qu8_st21_padVALID_kern22",
136 "max_pool_2d_1x8x4x17_qu8_st21_padVALID_kern21",
137 "max_pool_2d_1x32x32x8_qi8_st12_padVALID_kern22"
Jeremy Johnson88588622022-07-12 16:42:29 +0100138 ],
139 "profile": [
140 "tosa-bi",
141 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100142 ]
143 },
144 "minimum": {
145 "alternate_names": [
146 "min"
147 ],
148 "tests": [
149 "minimum_1x4x4x4_i32",
150 "minimum_64_i32"
Jeremy Johnson88588622022-07-12 16:42:29 +0100151 ],
152 "profile": [
153 "tosa-bi",
154 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100155 ]
156 },
157 "mul": {
158 "tests": [
159 "mul_13x21x3_qi16",
160 "mul_1x4x8x19_qu8"
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 "pack": {
168 "alternate_names": [
169 "stack"
170 ],
171 "tests": [
172 "pack_13x21x3_i32_axis0",
173 "pack_14x19_i32_axis1",
174 "pack_1x7x7x9_i32_axis2",
175 "pack_1x8x4x17_i32_axis4"
Jeremy Johnson88588622022-07-12 16:42:29 +0100176 ],
177 "profile": [
178 "tosa-bi",
179 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100180 ]
181 },
182 "reshape": {
183 "tests": [
184 "reshape_14x19_i32_rank1",
185 "reshape_13x21x3_i32_rank3",
186 "reshape_1x32x32x8_i32_rank2"
Jeremy Johnson88588622022-07-12 16:42:29 +0100187 ],
188 "profile": [
189 "tosa-bi",
190 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100191 ]
192 },
193 "resize_bilinear": {
Jeremy Johnsone81b2ba2023-03-13 12:21:22 +0000194 "alternate_names": [
195 "resize"
196 ],
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100197 "tests": [
Jeremy Johnsone81b2ba2023-03-13 12:21:22 +0000198 "resize_bilinear_1x32x32x8_qi16_bilinear_align0_half1_scale3",
199 "resize_bilinear_1x4x4x4_qi8_bilinear_align1_half0_scale1"
Jeremy Johnson88588622022-07-12 16:42:29 +0100200 ],
201 "profile": [
202 "tosa-bi",
203 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100204 ]
205 },
206 "resize_nearest_neighbor": {
207 "alternate_names": [
Jeremy Johnsone81b2ba2023-03-13 12:21:22 +0000208 "resize"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100209 ],
210 "tests": [
Jeremy Johnsone81b2ba2023-03-13 12:21:22 +0000211 "resize_nearest_neighbor_1x4x8x19_qu8_nearest_align0_half1_scale2",
212 "resize_nearest_neighbor_1x8x4x17_qi8_nearest_align0_half0_scale3"
Jeremy Johnson88588622022-07-12 16:42:29 +0100213 ],
214 "profile": [
215 "tosa-bi",
216 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100217 ]
218 },
219 "softmax": {
220 "tests": [
221 "softmax_1_qi8",
222 "softmax_14x19_qi16",
223 "softmax_1x32x32x8_qu8"
Jeremy Johnson88588622022-07-12 16:42:29 +0100224 ],
225 "profile": [
226 "tosa-bi",
227 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100228 ]
229 },
230 "split": {
231 "tests": [
Jeremy Johnsone81b2ba2023-03-13 12:21:22 +0000232 "split_3x1x1x7_i32_splitv_axis0",
Jeremy Johnson60136852023-02-09 12:16:14 +0000233 "split_13x21x3_i32_split3_axis1",
234 "split_1x4x8x19_i32_splitv_axis2",
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100235 "split_64_i32_split2_axis0"
Jeremy Johnson88588622022-07-12 16:42:29 +0100236 ],
237 "profile": [
238 "tosa-bi",
239 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100240 ]
241 },
242 "strided_slice": {
243 "tests": [
244 "strided_slice_13x21x3_i32_perm0",
245 "strided_slice_14x19_i32_perm1",
246 "strided_slice_64_i32_perm3"
Jeremy Johnson88588622022-07-12 16:42:29 +0100247 ],
248 "profile": [
249 "tosa-bi",
250 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100251 ]
252 },
253 "tanh": {
254 "tests": [
255 "tanh_13x21x3_qu8",
256 "tanh_14x19_qi8",
257 "tanh_1x8x4x17_qi16"
Jeremy Johnson88588622022-07-12 16:42:29 +0100258 ],
259 "profile": [
260 "tosa-bi",
261 "tosa-mi"
Jeremy Johnson0ecfa372022-06-30 14:27:56 +0100262 ]
263 }
264}