Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / b5fcfc0cfeffeb3fe9f05c32dc678ab09b2cee31 / . / reference_model / src / ops / op_factory.h
blob: f1d1680863b7b3d98e7caf5107fd5b22d2be85fb [file] [log] [blame]
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]1
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]2// Copyright (c) 2020-2023, 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#ifndef OPS_OP_FACTORY_H
17#define OPS_OP_FACTORY_H
18
19#include "attribute.h"
20#include "graph_node.h"
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]21#include "template_types.h"
22#include "tosa_serialization_handler.h"
23
24#define DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, RANK, DTYPE) \
25 case RANK: \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]26 return new OP<RANK, TOSA_REF_TYPE_##DTYPE>(sgt, attribute, id);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]27
28#define DEF_FACTORY_ONE_RANK_TWO_TYPE(OP, RANK, DTYPE1, DTYPE2) \
29 case RANK: \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]30 return new OP<RANK, TOSA_REF_TYPE_##DTYPE1, TOSA_REF_TYPE_##DTYPE2>(sgt, attribute, id);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]31
32#define DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, RANK1, RANK2, DTYPE) \
33 case RANK2: \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]34 return new OP<RANK1, RANK2, TOSA_REF_TYPE_##DTYPE>(sgt, attribute, id);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]35
36#define DEF_FACTORY_TWO_RANK_TWO_TYPE(OP, RANK1, RANK2, DTYPE1, DTYPE2) \
37 case RANK2: \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]38 return new OP<RANK1, RANK2, TOSA_REF_TYPE_##DTYPE1, TOSA_REF_TYPE_##DTYPE2>(sgt, attribute, id);
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]39
40#define DEF_FACTORY_ONE_RANK_0_6(OP) \
41 switch (inputRank) \
42 { \
43 case 0: \
TatWai Chongf7326092022-06-08 12:17:14 -0700[diff] [blame]44 return new OP<0>(sgt, attribute, id); \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]45 case 1: \
TatWai Chongf7326092022-06-08 12:17:14 -0700[diff] [blame]46 return new OP<1>(sgt, attribute, id); \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]47 case 2: \
TatWai Chongf7326092022-06-08 12:17:14 -0700[diff] [blame]48 return new OP<2>(sgt, attribute, id); \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]49 case 3: \
TatWai Chongf7326092022-06-08 12:17:14 -0700[diff] [blame]50 return new OP<3>(sgt, attribute, id); \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]51 case 4: \
TatWai Chongf7326092022-06-08 12:17:14 -0700[diff] [blame]52 return new OP<4>(sgt, attribute, id); \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]53 case 5: \
TatWai Chongf7326092022-06-08 12:17:14 -0700[diff] [blame]54 return new OP<5>(sgt, attribute, id); \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]55 case 6: \
TatWai Chongf7326092022-06-08 12:17:14 -0700[diff] [blame]56 return new OP<6>(sgt, attribute, id); \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]57 }
58
59#define DEF_FACTORY_ONE_TYPE(OP, DTYPE) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]60 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE) \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]61 { \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]62 return new OP<TOSA_REF_TYPE_##DTYPE>(sgt, attribute, id); \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]63 }
64
Tai Ly8ead6c42024-02-14 22:35:44 +0000[diff] [blame]65#define DEF_FACTORY_ONE_TYPE_ONE_ACCUM(OP, ATTR_NAME, DTYPE, ACC_TYPE) \
66 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE && ACCUM_FROM_ATTRIBUTE(ATTR_NAME) == TOSA_REF_TYPE_##ACC_TYPE) \
James Ward8b390432022-08-12 20:48:56 +0100[diff] [blame]67 { \
Tai Ly8ead6c42024-02-14 22:35:44 +0000[diff] [blame]68 return new OP<TOSA_REF_TYPE_##DTYPE, TOSA_REF_TYPE_##ACC_TYPE>(sgt, attribute, id); \
James Ward8b390432022-08-12 20:48:56 +0100[diff] [blame]69 }
70
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]71#define DEF_FACTORY_TWO_TYPE(OP, DTYPE1, DTYPE2) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]72 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE1 && weightDTYPE == TOSA_REF_TYPE_##DTYPE2) \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]73 { \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]74 return new OP<TOSA_REF_TYPE_##DTYPE1, TOSA_REF_TYPE_##DTYPE2>(sgt, attribute, id); \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]75 }
76
James Wardd34b3fc2023-01-18 14:51:25 +0000[diff] [blame]77#define DEF_FACTORY_TWO_TYPE_IN_OUT(OP, DTYPE1, DTYPE2) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]78 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE1 && outputDTYPE == TOSA_REF_TYPE_##DTYPE2) \
James Wardd34b3fc2023-01-18 14:51:25 +0000[diff] [blame]79 { \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]80 return new OP<TOSA_REF_TYPE_##DTYPE1, TOSA_REF_TYPE_##DTYPE2>(sgt, attribute, id); \
James Wardd34b3fc2023-01-18 14:51:25 +0000[diff] [blame]81 }
82
Tai Ly8ead6c42024-02-14 22:35:44 +0000[diff] [blame]83#define DEF_FACTORY_TWO_TYPE_ONE_ACCUM(OP, ATTR_NAME, DTYPE1, DTYPE2, ACC_TYPE) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]84 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE1 && weightDTYPE == TOSA_REF_TYPE_##DTYPE2 && \
Tai Ly8ead6c42024-02-14 22:35:44 +0000[diff] [blame]85 ACCUM_FROM_ATTRIBUTE(ATTR_NAME) == TOSA_REF_TYPE_##ACC_TYPE) \
James Ward8b390432022-08-12 20:48:56 +0100[diff] [blame]86 { \
Tai Ly8ead6c42024-02-14 22:35:44 +0000[diff] [blame]87 return new OP<TOSA_REF_TYPE_##DTYPE1, TOSA_REF_TYPE_##DTYPE2, TOSA_REF_TYPE_##ACC_TYPE>(sgt, attribute, id); \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]88 }
James Ward8b390432022-08-12 20:48:56 +0100[diff] [blame]89
James Wardd34b3fc2023-01-18 14:51:25 +0000[diff] [blame]90#define DEF_FACTORY_THREE_TYPE(OP, DTYPE1, DTYPE2, DTYPE3) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]91 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE1 && weightDTYPE == TOSA_REF_TYPE_##DTYPE2 && \
92 outputDTYPE == TOSA_REF_TYPE_##DTYPE3) \
James Wardd34b3fc2023-01-18 14:51:25 +0000[diff] [blame]93 { \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]94 return new OP<TOSA_REF_TYPE_##DTYPE1, TOSA_REF_TYPE_##DTYPE2, TOSA_REF_TYPE_##DTYPE3>(sgt, attribute, id); \
James Wardd34b3fc2023-01-18 14:51:25 +0000[diff] [blame]95 }
96
Tai Lyf36f2562024-03-14 16:21:29 +0000[diff] [blame]97#define DEF_FACTORY_THREE_TYPE_ONE_ACCUM(OP, ATTR_NAME, IN_DTYPE, W_DTYPE, ACC_DTYPE, OUT_DTYPE) \
98 if (inputDTYPE == TOSA_REF_TYPE_##IN_DTYPE && weightDTYPE == TOSA_REF_TYPE_##W_DTYPE && \
99 outputDTYPE == TOSA_REF_TYPE_##OUT_DTYPE && ACCUM_FROM_ATTRIBUTE(ATTR_NAME) == TOSA_REF_TYPE_##ACC_DTYPE) \
100 { \
101 return new OP<TOSA_REF_TYPE_##IN_DTYPE, TOSA_REF_TYPE_##W_DTYPE, TOSA_REF_TYPE_##ACC_DTYPE, \
102 TOSA_REF_TYPE_##OUT_DTYPE>(sgt, attribute, id); \
103 }
104
James Ward8b390432022-08-12 20:48:56 +0100[diff] [blame]105// Statement-expression to evaluate accumulate attribute in-place
106#define ACCUM_FROM_ATTRIBUTE(ATTRIBUTE_NAME) \
107 ({ \
108 tosa::DType accumDType = tosa::DType_UNKNOWN; \
109 if (auto p = dynamic_cast<tosa::Tosa##ATTRIBUTE_NAME##Attribute*>(attribute)) \
110 { \
111 auto attr = new tosa::Tosa##ATTRIBUTE_NAME##Attribute(p); \
112 ASSERT_MEM(attr); \
Tai Ly8ead6c42024-02-14 22:35:44 +0000[diff] [blame]113 accumDType = tosa::EnumValuesDType()[attr->acc_type()]; \
James Ward8b390432022-08-12 20:48:56 +0100[diff] [blame]114 } \
115 else \
116 { \
117 FATAL_ERROR("Can't initialize Tosa" #ATTRIBUTE_NAME "Attribute.\nPre-initialization " \
118 "of this attribute is required in order to determine the accumulate type."); \
119 } \
Jerry Ge9c9c8da2023-07-19 23:08:16 +0000[diff] [blame]120 ConvertDType(accumDType); \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]121 })
James Ward8b390432022-08-12 20:48:56 +0100[diff] [blame]122
TatWai Chongf7326092022-06-08 12:17:14 -0700[diff] [blame]123#define DEF_FACTORY_TWO_TYPE_RESIZE_INT16(OP, DTYPE1, DTYPE2) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]124 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE1 && outputDTYPE == TOSA_REF_TYPE_##DTYPE2) \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]125 { \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]126 return new OP<TOSA_REF_TYPE_##DTYPE1, TOSA_REF_TYPE_##DTYPE2, int16_t>(sgt, attribute, id); \
TatWai Chongf7326092022-06-08 12:17:14 -0700[diff] [blame]127 }
128
James Ward8b390432022-08-12 20:48:56 +0100[diff] [blame]129#define DEF_FACTORY_TWO_TYPE_RESIZE_FP16(OP, DTYPE1, DTYPE2) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]130 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE1 && outputDTYPE == TOSA_REF_TYPE_##DTYPE2) \
James Ward8b390432022-08-12 20:48:56 +0100[diff] [blame]131 { \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]132 return new OP<TOSA_REF_TYPE_##DTYPE1, TOSA_REF_TYPE_##DTYPE2, half_float::half>(sgt, attribute, id); \
James Ward8b390432022-08-12 20:48:56 +0100[diff] [blame]133 }
134
James Ward24dbc422022-10-19 12:20:31 +0100[diff] [blame]135#define DEF_FACTORY_TWO_TYPE_RESIZE_BF16(OP, DTYPE1, DTYPE2) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]136 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE1 && outputDTYPE == TOSA_REF_TYPE_##DTYPE2) \
James Ward24dbc422022-10-19 12:20:31 +0100[diff] [blame]137 { \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]138 return new OP<TOSA_REF_TYPE_##DTYPE1, TOSA_REF_TYPE_##DTYPE2, Eigen::bfloat16>(sgt, attribute, id); \
James Ward24dbc422022-10-19 12:20:31 +0100[diff] [blame]139 }
140
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]141#define DEF_FACTORY_TWO_TYPE_RESIZE_FP32(OP, DTYPE1, DTYPE2) \
142 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE1 && outputDTYPE == TOSA_REF_TYPE_##DTYPE2) \
TatWai Chongf7326092022-06-08 12:17:14 -0700[diff] [blame]143 { \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]144 return new OP<TOSA_REF_TYPE_##DTYPE1, TOSA_REF_TYPE_##DTYPE2, float>(sgt, attribute, id); \
145 }
146
147#define DEF_FACTORY_TWO_TYPE_RESIZE_FP64(OP, DTYPE1, DTYPE2) \
148 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE1 && outputDTYPE == TOSA_REF_TYPE_##DTYPE2) \
149 { \
150 return new OP<TOSA_REF_TYPE_##DTYPE1, TOSA_REF_TYPE_##DTYPE2, double>(sgt, attribute, id); \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]151 }
152
153#define DEF_FACTORY_RANK0_6_ONE_RANK_ONE_TYPE(OP, DTYPE) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]154 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE) \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]155 { \
156 switch (inputRank) \
157 { \
158 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 0, DTYPE) \
159 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 1, DTYPE) \
160 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 2, DTYPE) \
161 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 3, DTYPE) \
162 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 4, DTYPE) \
163 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 5, DTYPE) \
164 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 6, DTYPE) \
165 } \
166 }
167
168#define DEF_FACTORY_RANK1_6_ONE_RANK_ONE_TYPE(OP, DTYPE) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]169 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE) \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]170 { \
171 switch (inputRank) \
172 { \
173 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 1, DTYPE) \
174 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 2, DTYPE) \
175 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 3, DTYPE) \
176 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 4, DTYPE) \
177 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 5, DTYPE) \
178 DEF_FACTORY_ONE_RANK_ONE_TYPE(OP, 6, DTYPE) \
179 } \
180 }
181
182#define DEF_FACTORY_RANK0_6_ONE_RANK_TWO_TYPE(OP, DTYPE1, DTYPE2) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]183 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE1 && outputDTYPE == TOSA_REF_TYPE_##DTYPE2) \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]184 { \
185 switch (inputRank) \
186 { \
187 DEF_FACTORY_ONE_RANK_TWO_TYPE(OP, 0, DTYPE1, DTYPE2) \
188 DEF_FACTORY_ONE_RANK_TWO_TYPE(OP, 1, DTYPE1, DTYPE2) \
189 DEF_FACTORY_ONE_RANK_TWO_TYPE(OP, 2, DTYPE1, DTYPE2) \
190 DEF_FACTORY_ONE_RANK_TWO_TYPE(OP, 3, DTYPE1, DTYPE2) \
191 DEF_FACTORY_ONE_RANK_TWO_TYPE(OP, 4, DTYPE1, DTYPE2) \
192 DEF_FACTORY_ONE_RANK_TWO_TYPE(OP, 5, DTYPE1, DTYPE2) \
193 DEF_FACTORY_ONE_RANK_TWO_TYPE(OP, 6, DTYPE1, DTYPE2) \
194 } \
195 }
196
197#define DEF_FACTORY_RESHAPE(OP, DTYPE) \
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]198 if (inputDTYPE == TOSA_REF_TYPE_##DTYPE && outputDTYPE == TOSA_REF_TYPE_##DTYPE) \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]199 { \
200 switch (inputRank) \
201 { \
Jerry Ge9c9c8da2023-07-19 23:08:16 +0000[diff] [blame]202 case 0: { \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]203 switch (outputRank) \
204 { \
205 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 0, 0, DTYPE) \
206 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 0, 1, DTYPE) \
207 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 0, 2, DTYPE) \
208 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 0, 3, DTYPE) \
209 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 0, 4, DTYPE) \
210 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 0, 5, DTYPE) \
211 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 0, 6, DTYPE) \
212 } \
213 } \
Jerry Ge9c9c8da2023-07-19 23:08:16 +0000[diff] [blame]214 case 1: { \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]215 switch (outputRank) \
216 { \
217 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 1, 0, DTYPE) \
218 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 1, 1, DTYPE) \
219 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 1, 2, DTYPE) \
220 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 1, 3, DTYPE) \
221 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 1, 4, DTYPE) \
222 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 1, 5, DTYPE) \
223 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 1, 6, DTYPE) \
224 } \
225 } \
Jerry Ge9c9c8da2023-07-19 23:08:16 +0000[diff] [blame]226 case 2: { \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]227 switch (outputRank) \
228 { \
229 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 2, 0, DTYPE) \
230 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 2, 1, DTYPE) \
231 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 2, 2, DTYPE) \
232 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 2, 3, DTYPE) \
233 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 2, 4, DTYPE) \
234 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 2, 5, DTYPE) \
235 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 2, 6, DTYPE) \
236 } \
237 } \
Jerry Ge9c9c8da2023-07-19 23:08:16 +0000[diff] [blame]238 case 3: { \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]239 switch (outputRank) \
240 { \
241 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 3, 0, DTYPE) \
242 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 3, 1, DTYPE) \
243 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 3, 2, DTYPE) \
244 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 3, 3, DTYPE) \
245 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 3, 4, DTYPE) \
246 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 3, 5, DTYPE) \
247 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 3, 6, DTYPE) \
248 } \
249 } \
Jerry Ge9c9c8da2023-07-19 23:08:16 +0000[diff] [blame]250 case 4: { \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]251 switch (outputRank) \
252 { \
253 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 4, 0, DTYPE) \
254 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 4, 1, DTYPE) \
255 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 4, 2, DTYPE) \
256 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 4, 3, DTYPE) \
257 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 4, 4, DTYPE) \
258 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 4, 5, DTYPE) \
259 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 4, 6, DTYPE) \
260 } \
261 } \
Jerry Ge9c9c8da2023-07-19 23:08:16 +0000[diff] [blame]262 case 5: { \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]263 switch (outputRank) \
264 { \
265 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 5, 0, DTYPE) \
266 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 5, 1, DTYPE) \
267 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 5, 2, DTYPE) \
268 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 5, 3, DTYPE) \
269 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 5, 4, DTYPE) \
270 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 5, 5, DTYPE) \
271 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 5, 6, DTYPE) \
272 } \
273 } \
Jerry Ge9c9c8da2023-07-19 23:08:16 +0000[diff] [blame]274 case 6: { \
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]275 switch (outputRank) \
276 { \
277 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 6, 0, DTYPE) \
278 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 6, 1, DTYPE) \
279 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 6, 2, DTYPE) \
280 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 6, 3, DTYPE) \
281 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 6, 4, DTYPE) \
282 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 6, 5, DTYPE) \
283 DEF_FACTORY_TWO_RANK_ONE_TYPE(OP, 6, 6, DTYPE) \
284 } \
285 } \
286 } \
287 }
288
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]289namespace TosaReference
290{
291
Kevin Chengacb550f2021-06-29 15:32:19 -0700[diff] [blame]292class SubgraphTraverser;
293class GraphNode;
294
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]295class OpFactory
296{
297public:
Kevin Chengacb550f2021-06-29 15:32:19 -0700[diff] [blame]298 static GraphNode* newOp(SubgraphTraverser* sgt,
299 tosa::TosaSerializationHandler* tsh,
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]300 tosa::Op opType,
301 tosa::TosaAttributeBase* attribute,
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]302 uint64_t id,
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]303 TOSA_REF_TYPE inputDTYPE,
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]304 int inputRank,
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]305 TOSA_REF_TYPE outputDTYPE,
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]306 int outputRank,
Tai Lya4d748b2023-03-28 22:06:56 +0000[diff] [blame]307 TOSA_REF_TYPE weightDTYPE,
Eric Kunzee5e26762020-10-13 16:11:07 -0700[diff] [blame]308 int weightRank);
309};
310}; // namespace TosaReference
311
312#endif