MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / BuiltinOpResolver

Method BuiltinOpResolver

tensorflow/lite/kernels/register.cc:158–397  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

156}
157
158BuiltinOpResolver::BuiltinOpResolver() {
159 AddBuiltin(BuiltinOperator_ABS, Register_ABS());
160 AddBuiltin(BuiltinOperator_HARD_SWISH, Register_HARD_SWISH());
161 AddBuiltin(BuiltinOperator_RELU, Register_RELU());
162 AddBuiltin(BuiltinOperator_RELU_N1_TO_1, Register_RELU_N1_TO_1());
163 AddBuiltin(BuiltinOperator_RELU6, Register_RELU6(), /* min_version */ 1,
164 /* max_version */ 2);
165 AddBuiltin(BuiltinOperator_TANH, Register_TANH(), /* min_version */ 1,
166 /* max_version */ 2);
167 AddBuiltin(BuiltinOperator_LOGISTIC, Register_LOGISTIC(),
168 /* min_version */ 1,
169 /* max_version */ 2);
170 AddBuiltin(BuiltinOperator_AVERAGE_POOL_2D, Register_AVERAGE_POOL_2D(),
171 /* min_version */ 1,
172 /* max_version */ 2);
173 AddBuiltin(BuiltinOperator_MAX_POOL_2D, Register_MAX_POOL_2D(),
174 /* min_version */ 1,
175 /* max_version */ 2);
176 AddBuiltin(BuiltinOperator_L2_POOL_2D, Register_L2_POOL_2D());
177 AddBuiltin(BuiltinOperator_CONV_2D, Register_CONV_2D(),
178 /* min_version */ 1,
179 /* max_version */ 3);
180 AddBuiltin(BuiltinOperator_DEPTHWISE_CONV_2D, Register_DEPTHWISE_CONV_2D(),
181 /* min_version */ 1,
182 /* max_version */ 3);
183 AddBuiltin(BuiltinOperator_SVDF, Register_SVDF(),
184 /* min_version */ 1,
185 /* max_version */ 2);
186 AddBuiltin(BuiltinOperator_RNN, Register_RNN(),
187 /* min_version */ 1,
188 /* max_version */ 2);
189 AddBuiltin(BuiltinOperator_BIDIRECTIONAL_SEQUENCE_RNN,
190 Register_BIDIRECTIONAL_SEQUENCE_RNN(),
191 /* min_version */ 1,
192 /* max_version */ 2);
193 AddBuiltin(BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_RNN,
194 Register_UNIDIRECTIONAL_SEQUENCE_RNN(),
195 /* min_version */ 1,
196 /* max_version */ 2);
197 AddBuiltin(BuiltinOperator_EMBEDDING_LOOKUP, Register_EMBEDDING_LOOKUP(),
198 /* min_version */ 1,
199 /* max_version */ 3);
200 AddBuiltin(BuiltinOperator_EMBEDDING_LOOKUP_SPARSE,
201 Register_EMBEDDING_LOOKUP_SPARSE());
202 AddBuiltin(BuiltinOperator_FULLY_CONNECTED, Register_FULLY_CONNECTED(),
203 /* min_version */ 1,
204 /* max_version */ 6);
205 AddBuiltin(BuiltinOperator_LSH_PROJECTION, Register_LSH_PROJECTION());
206 AddBuiltin(BuiltinOperator_HASHTABLE_LOOKUP, Register_HASHTABLE_LOOKUP());
207 AddBuiltin(BuiltinOperator_SOFTMAX, Register_SOFTMAX(),
208 /* min_version */ 1,
209 /* max_version */ 2);
210 AddBuiltin(BuiltinOperator_CONCATENATION, Register_CONCATENATION(),
211 /* min_version */ 1,
212 /* max_version */ 2);
213 AddBuiltin(BuiltinOperator_ADD, Register_ADD(),
214 /* min_version */ 1,
215 /* max_version */ 2);

Callers

nothing calls this directly

Calls 15

Register_HARD_SWISHFunction · 0.85
Register_RELUFunction · 0.85
Register_RELU_N1_TO_1Function · 0.85
Register_RELU6Function · 0.85
Register_TANHFunction · 0.85
Register_LOGISTICFunction · 0.85
Register_L2_POOL_2DFunction · 0.85
Register_SVDFFunction · 0.85
Register_RNNFunction · 0.85

Tested by

no test coverage detected