Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / 14d7f7a2b5d0d85b83d8c84a5456828feb1a0ea1 / . / reference_model / src / ops / op_factory.cc
blob: 440d624b69cac87d7dece79625ecfea449fed1e5 [file] [log] [blame]
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]1
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]2// Copyright (c) 2020-2021, ARM Limited.
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]3//
4// Licensed under the Apache License, Version 2.0 (the "License");
5// you may not use this file except in compliance with the License.
6// You may obtain a copy of the License at
7//
8// http://www.apache.org/licenses/LICENSE-2.0
9//
10// Unless required by applicable law or agreed to in writing, software
11// distributed under the License is distributed on an "AS IS" BASIS,
12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13// See the License for the specific language governing permissions and
14// limitations under the License.
15
16#include "op_factory.h"
17#include "activation_funcs.h"
18#include "comparison.h"
19#include "control_flow.h"
20#include "custom.h"
21#include "data_layout.h"
22#include "data_nodes.h"
23#include "ewise_binary.h"
24#include "ewise_ternary.h"
25#include "ewise_unary.h"
26#include "image.h"
27#include "reduction.h"
28#include "scatter_gather.h"
29#include "tensor_ops.h"
30#include "type_conversion.h"
31
32using namespace TosaReference;
33using namespace tosa;
34
35GraphNode* OpFactory::newOp(TosaSerializationHandler* tsh,
36 Op opType,
37 TosaAttributeBase* attribute,
38 TosaQuantInfoBase* qinfo,
39 uint64_t id,
40 DType inputDType,
41 int inputRank,
42 DType outputDType,
43 int outputRank,
44 DType weightDType,
45 int weightRank)
46{
47 switch (opType)
48 {
49 // tensor_ops
50 case Op_ARGMAX:
51 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpArgMax, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]52 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpArgMax, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]53 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpArgMax, INT16);
54 break;
55 case Op_AVG_POOL2D:
56 DEF_FACTORY_ONE_TYPE(OpAvgPool2d, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]57 DEF_FACTORY_ONE_TYPE(OpAvgPool2d, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]58 DEF_FACTORY_ONE_TYPE(OpAvgPool2d, INT16);
59 break;
60 case Op_CONV2D:
61 DEF_FACTORY_TWO_TYPE(OpConv2d, FLOAT, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]62 DEF_FACTORY_TWO_TYPE(OpConv2d, INT8, INT4);
63 DEF_FACTORY_TWO_TYPE(OpConv2d, INT8, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]64 DEF_FACTORY_TWO_TYPE(OpConv2d, INT16, INT8);
65 break;
66 case Op_DEPTHWISE_CONV2D:
67 DEF_FACTORY_TWO_TYPE(OpDepthwiseConv2d, FLOAT, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]68 DEF_FACTORY_TWO_TYPE(OpDepthwiseConv2d, INT8, INT4);
69 DEF_FACTORY_TWO_TYPE(OpDepthwiseConv2d, INT8, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]70 DEF_FACTORY_TWO_TYPE(OpDepthwiseConv2d, INT16, INT8);
71 break;
72 case Op_FULLY_CONNECTED:
73 DEF_FACTORY_TWO_TYPE(OpFullyConnected, FLOAT, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]74 DEF_FACTORY_TWO_TYPE(OpFullyConnected, INT8, INT4);
75 DEF_FACTORY_TWO_TYPE(OpFullyConnected, INT8, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]76 DEF_FACTORY_TWO_TYPE(OpFullyConnected, INT16, INT8);
77 break;
78 case Op_MATMUL:
79 DEF_FACTORY_ONE_TYPE(OpMatMul, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]80 DEF_FACTORY_ONE_TYPE(OpMatMul, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]81 DEF_FACTORY_ONE_TYPE(OpMatMul, INT16);
82 break;
83 case Op_MAX_POOL2D:
84 DEF_FACTORY_ONE_TYPE(OpMaxPool2d, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]85 DEF_FACTORY_ONE_TYPE(OpMaxPool2d, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]86 DEF_FACTORY_ONE_TYPE(OpMaxPool2d, INT16);
87 break;
88 case Op_TRANSPOSE_CONV2D:
89 DEF_FACTORY_TWO_TYPE(OpTransposeConv2d, FLOAT, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]90 DEF_FACTORY_TWO_TYPE(OpTransposeConv2d, INT8, INT4);
91 DEF_FACTORY_TWO_TYPE(OpTransposeConv2d, INT8, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]92 DEF_FACTORY_TWO_TYPE(OpTransposeConv2d, INT16, INT8);
93 break;
94
95 // activation_funcs
96 case Op_CLAMP:
97 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpClamp, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]98 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpClamp, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]99 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpClamp, INT16);
100 break;
101 case Op_RELUN:
102 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpReluN, FLOAT);
103 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpReluN, INT32);
104 break;
105 case Op_SIGMOID:
106 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSigmoid, FLOAT);
107 break;
108 case Op_TANH:
109 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTanh, FLOAT);
110 break;
111
112 // ewise_binary
113 case Op_ADD:
114 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpAdd, FLOAT);
115 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpAdd, INT32);
116 break;
117 case Op_ARITHMETIC_RIGHT_SHIFT:
118 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpArithmeticRightShift, INT8);
119 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpArithmeticRightShift, INT16);
120 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpArithmeticRightShift, INT32);
121 break;
122 case Op_BITWISE_AND:
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]123 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseAnd, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]124 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseAnd, INT16);
125 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseAnd, INT32);
126 break;
127 case Op_BITWISE_OR:
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]128 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseOr, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]129 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseOr, INT16);
130 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseOr, INT32);
131 break;
132 case Op_BITWISE_XOR:
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]133 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseXor, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]134 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseXor, INT16);
135 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseXor, INT32);
136 break;
Kevin Cheng14d7f7a2021-05-12 10:44:49 -0700[diff] [blame^]137 case Op_DIV:
138 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpDiv, INT32);
139 break;
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]140 case Op_LOGICAL_AND:
141 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpLogicalAnd, BOOL);
142 break;
143 case Op_LOGICAL_LEFT_SHIFT:
144 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpLogicalLeftShift, INT8);
145 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpLogicalLeftShift, INT16);
146 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpLogicalLeftShift, INT32);
147 break;
148 case Op_LOGICAL_RIGHT_SHIFT:
149 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpLogicalRightShift, INT8);
150 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpLogicalRightShift, INT16);
151 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpLogicalRightShift, INT32);
152 break;
153 case Op_LOGICAL_OR:
154 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpLogicalOr, BOOL);
155 break;
156 case Op_LOGICAL_XOR:
157 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpLogicalOr, BOOL);
158 break;
159 case Op_MAXIMUM:
160 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpMaximum, FLOAT);
161 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpMaximum, INT32);
162 break;
163 case Op_MINIMUM:
164 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpMinimum, FLOAT);
165 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpMinimum, INT32);
166 break;
167 case Op_MUL:
168 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpMul, FLOAT, FLOAT);
169 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpMul, INT8, INT32);
170 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpMul, INT16, INT32);
171 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpMul, INT32, INT32);
172 break;
173 case Op_POW:
174 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpPow, FLOAT);
175 break;
176 case Op_SUB:
177 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSub, FLOAT);
178 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSub, INT32);
179 break;
180 case Op_TABLE:
181 DEF_FACTORY_ONE_RANK_0_6(OpTable);
182 break;
183
184 // ewise_unary
185 case Op_ABS:
186 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpAbs, FLOAT);
187 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpAbs, INT32);
188 break;
189 case Op_BITWISE_NOT:
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]190 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseNot, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]191 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseNot, INT16);
192 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpBitwiseNot, INT32);
193 break;
194 case Op_CEIL:
195 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpCeil, FLOAT);
196 break;
197 case Op_CLZ:
198 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpClz, INT32);
199 break;
200 case Op_EXP:
201 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpExp, FLOAT);
202 break;
203 case Op_FLOOR:
204 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpFloor, FLOAT);
205 break;
206 case Op_LOG:
207 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpLog, FLOAT);
208 break;
209 case Op_LOGICAL_NOT:
210 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpLogicalNot, BOOL);
211 break;
212 case Op_NEGATE:
213 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpNegate, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]214 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpNegate, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]215 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpNegate, INT16);
216 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpNegate, INT32);
217 break;
218 case Op_RECIPROCAL:
219 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpReciprocal, FLOAT);
220 break;
221 case Op_RSQRT:
222 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpRsqrt, FLOAT);
223 break;
224
225 // ewise_ternary
226 case Op_SELECT:
227 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSelect, FLOAT);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]228 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSelect, INT8);
229 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSelect, INT16);
230 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSelect, INT32);
231 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSelect, BOOL);
232 break;
233
234 // comparison
235 case Op_EQUAL:
236 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpEqual, FLOAT);
237 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpEqual, INT32);
238 break;
239 case Op_GREATER:
240 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpGreater, FLOAT);
241 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpGreater, INT32);
242 break;
243 case Op_GREATER_EQUAL:
244 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpGreaterEqual, FLOAT);
245 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpGreaterEqual, INT32);
246 break;
247
248 // reduction
249 case Op_REDUCE_ALL:
250 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceAll, BOOL);
251 break;
252 case Op_REDUCE_ANY:
253 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceAny, BOOL);
254 break;
255 case Op_REDUCE_MAX:
256 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceMax, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]257 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceMax, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]258 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceMax, INT16);
259 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceMax, INT32);
260 break;
261 case Op_REDUCE_MIN:
262 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceMin, FLOAT);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]263 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceMin, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]264 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceMin, INT16);
265 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceMin, INT32);
266 break;
267 case Op_REDUCE_PRODUCT:
268 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceProduct, FLOAT);
269 break;
270 case Op_REDUCE_SUM:
271 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceSum, FLOAT);
272 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceSum, INT32);
273 break;
274
275 // data layout
276 case Op_CONCAT:
277 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpConcat, FLOAT);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]278 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpConcat, INT8);
279 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpConcat, INT16);
280 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpConcat, INT32);
281 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpConcat, BOOL);
282 break;
283 case Op_PAD:
284 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpPad, FLOAT);
285 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpPad, INT32);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]286 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpPad, INT8);
287 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpPad, INT16);
288 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpPad, BOOL);
289 break;
290 case Op_RESHAPE:
291 DEF_FACTORY_RESHAPE(OpReshape, FLOAT);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]292 DEF_FACTORY_RESHAPE(OpReshape, INT8);
293 DEF_FACTORY_RESHAPE(OpReshape, INT16);
294 DEF_FACTORY_RESHAPE(OpReshape, INT32);
295 DEF_FACTORY_RESHAPE(OpReshape, BOOL);
296 break;
297 case Op_REVERSE:
298 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReverse, FLOAT);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]299 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReverse, INT8);
300 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReverse, INT16);
301 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReverse, INT32);
302 DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OpReverse, BOOL);
303 break;
304 case Op_SLICE:
305 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, FLOAT);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]306 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, INT8);
307 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, INT16);
308 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, INT32);
309 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpSlice, BOOL);
310 break;
311 case Op_TILE:
312 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, FLOAT);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]313 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, INT8);
314 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, INT16);
315 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, INT32);
316 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTile, BOOL);
317 break;
318 case Op_TRANSPOSE:
319 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, BOOL);
320 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, FLOAT);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]321 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, INT8);
322 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, INT16);
323 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpTranspose, INT32);
324 break;
325
326 // scatter_gather
327 case Op_GATHER:
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]328 DEF_FACTORY_ONE_TYPE(OpGather, INT8);
Kevin Cheng77d0f762020-11-24 10:26:32 -0800[diff] [blame]329 DEF_FACTORY_ONE_TYPE(OpGather, INT16);
330 DEF_FACTORY_ONE_TYPE(OpGather, INT32);
331 DEF_FACTORY_ONE_TYPE(OpGather, FLOAT);
332 break;
333 case Op_SCATTER:
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]334 DEF_FACTORY_ONE_TYPE(OpScatter, INT8);
Kevin Cheng77d0f762020-11-24 10:26:32 -0800[diff] [blame]335 DEF_FACTORY_ONE_TYPE(OpScatter, INT16);
336 DEF_FACTORY_ONE_TYPE(OpScatter, INT32);
337 DEF_FACTORY_ONE_TYPE(OpScatter, FLOAT);
338 break;
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]339
340 // image
341 case Op_RESIZE:
342 DEF_FACTORY_TWO_TYPE_RESIZE(OpResize, INT8, INT32);
343 DEF_FACTORY_TWO_TYPE_RESIZE(OpResize, INT8, INT8);
344 DEF_FACTORY_TWO_TYPE_RESIZE(OpResize, INT16, INT48);
345 DEF_FACTORY_TWO_TYPE_RESIZE(OpResize, INT16, INT16);
Kevin Cheng77d0f762020-11-24 10:26:32 -0800[diff] [blame]346 DEF_FACTORY_TWO_TYPE_RESIZE(OpResize, FLOAT, FLOAT);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]347 break;
348
349 // data_nodes
350 case Op_CONST:
351 return new OpConst(id);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]352 case Op_IDENTITY:
353 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, FLOAT);
354 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, INT32);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]355 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, INT8);
356 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, INT16);
357 DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OpIdentity, BOOL);
358 break;
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]359
360 // type_conversion
361 case Op_CAST:
362 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, BOOL, INT8);
363 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, BOOL, INT16);
364 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, BOOL, INT32);
365 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT8, BOOL);
366 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT8, INT16);
367 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT8, INT32);
368 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT8, FLOAT);
369 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT16, BOOL);
370 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT16, INT8);
371 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT16, INT32);
372 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT16, FLOAT);
373 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT32, BOOL);
374 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT32, INT8);
375 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT32, INT16);
376 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, INT32, FLOAT);
377 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, FLOAT, INT8);
378 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, FLOAT, INT16);
379 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpCast, FLOAT, INT32);
380 break;
381 case Op_RESCALE:
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]382 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT8, INT8);
383 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT8, INT16);
384 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT8, INT32);
385 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT16, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]386 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT16, INT16);
387 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT16, INT32);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]388 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT32, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]389 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT32, INT16);
390 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT32, INT32);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]391 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT48, INT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]392 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT48, INT16);
393 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT48, INT32);
Kevin Cheng3a478572021-01-22 17:21:02 -0800[diff] [blame]394 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, UINT8, INT8);
395 DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OpRescale, INT8, UINT8);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]396 break;
397
398 // custom
399 case Op_CUSTOM:
400 return new OpCustom(id);
401
402 // control_flow
403 case Op_COND_IF:
404 return new OpCondIf(tsh, attribute, id);
405 case Op_WHILE_LOOP:
406 return new OpWhileLoop(tsh, attribute, id);
407
408 // Ops not recognized
409 default:
410 goto done;
411
412 } // End of switch(opType)
413
414done:
415 return nullptr;
416}