Lines Matching full:resolver
109 void RegisterSelectedOps(tflite::MutableOpResolver* resolver) { in RegisterSelectedOps() argument
110 resolver->AddBuiltin(tflite::BuiltinOperator_ADD, in RegisterSelectedOps()
114 resolver->AddBuiltin(tflite::BuiltinOperator_CONCATENATION, in RegisterSelectedOps()
118 resolver->AddBuiltin(tflite::BuiltinOperator_CONV_2D, in RegisterSelectedOps()
122 resolver->AddBuiltin(tflite::BuiltinOperator_DEPTHWISE_CONV_2D, in RegisterSelectedOps()
126 resolver->AddBuiltin(tflite::BuiltinOperator_AVERAGE_POOL_2D, in RegisterSelectedOps()
130 resolver->AddBuiltin(::tflite::BuiltinOperator_EQUAL, in RegisterSelectedOps()
133 resolver->AddBuiltin(tflite::BuiltinOperator_FULLY_CONNECTED, in RegisterSelectedOps()
137 resolver->AddBuiltin(::tflite::BuiltinOperator_GREATER_EQUAL, in RegisterSelectedOps()
139 resolver->AddBuiltin(tflite::BuiltinOperator_L2_NORMALIZATION, in RegisterSelectedOps()
143 resolver->AddBuiltin(tflite::BuiltinOperator_MUL, in RegisterSelectedOps()
145 resolver->AddBuiltin(tflite::BuiltinOperator_RESHAPE, in RegisterSelectedOps()
147 resolver->AddBuiltin(::tflite::BuiltinOperator_REDUCE_MAX, in RegisterSelectedOps()
149 resolver->AddBuiltin(::tflite::BuiltinOperator_REDUCE_MIN, in RegisterSelectedOps()
151 resolver->AddBuiltin(::tflite::BuiltinOperator_REDUCE_ANY, in RegisterSelectedOps()
153 resolver->AddBuiltin(tflite::BuiltinOperator_SOFTMAX, in RegisterSelectedOps()
157 resolver->AddBuiltin(tflite::BuiltinOperator_GATHER, in RegisterSelectedOps()
161 resolver->AddBuiltin(::tflite::BuiltinOperator_GATHER_ND, in RegisterSelectedOps()
164 resolver->AddBuiltin(::tflite::BuiltinOperator_IF, in RegisterSelectedOps()
166 resolver->AddBuiltin(::tflite::BuiltinOperator_ROUND, in RegisterSelectedOps()
168 resolver->AddBuiltin(::tflite::BuiltinOperator_ZEROS_LIKE, in RegisterSelectedOps()
170 resolver->AddBuiltin(tflite::BuiltinOperator_TRANSPOSE, in RegisterSelectedOps()
174 resolver->AddBuiltin(tflite::BuiltinOperator_SUB, in RegisterSelectedOps()
178 resolver->AddBuiltin(tflite::BuiltinOperator_DIV, in RegisterSelectedOps()
180 resolver->AddBuiltin(tflite::BuiltinOperator_STRIDED_SLICE, in RegisterSelectedOps()
184 resolver->AddBuiltin(tflite::BuiltinOperator_EXP, in RegisterSelectedOps()
186 resolver->AddBuiltin(tflite::BuiltinOperator_TOPK_V2, in RegisterSelectedOps()
190 resolver->AddBuiltin(tflite::BuiltinOperator_SLICE, in RegisterSelectedOps()
194 resolver->AddBuiltin(tflite::BuiltinOperator_SPLIT, in RegisterSelectedOps()
198 resolver->AddBuiltin(tflite::BuiltinOperator_CAST, in RegisterSelectedOps()
200 resolver->AddBuiltin(tflite::BuiltinOperator_MAXIMUM, in RegisterSelectedOps()
204 resolver->AddBuiltin(tflite::BuiltinOperator_MINIMUM, in RegisterSelectedOps()
208 resolver->AddBuiltin(tflite::BuiltinOperator_NEG, in RegisterSelectedOps()
210 resolver->AddBuiltin(tflite::BuiltinOperator_SLICE, in RegisterSelectedOps()
214 resolver->AddBuiltin(tflite::BuiltinOperator_LOG, in RegisterSelectedOps()
216 resolver->AddBuiltin(tflite::BuiltinOperator_LOGISTIC, in RegisterSelectedOps()
218 resolver->AddBuiltin(tflite::BuiltinOperator_SUM, in RegisterSelectedOps()
220 resolver->AddBuiltin(tflite::BuiltinOperator_PACK, in RegisterSelectedOps()
224 resolver->AddBuiltin(tflite::BuiltinOperator_DEQUANTIZE, in RegisterSelectedOps()
228 resolver->AddBuiltin(tflite::BuiltinOperator_MEAN, in RegisterSelectedOps()
230 resolver->AddBuiltin(tflite::BuiltinOperator_LESS, in RegisterSelectedOps()
232 resolver->AddBuiltin(tflite::BuiltinOperator_TILE, in RegisterSelectedOps()
234 resolver->AddBuiltin(tflite::BuiltinOperator_SQUARED_DIFFERENCE, in RegisterSelectedOps()
236 resolver->AddBuiltin(tflite::BuiltinOperator_RSQRT, in RegisterSelectedOps()
238 resolver->AddBuiltin(tflite::BuiltinOperator_LOG_SOFTMAX, in RegisterSelectedOps()
240 resolver->AddBuiltin(::tflite::BuiltinOperator_WHERE, in RegisterSelectedOps()
242 resolver->AddBuiltin(tflite::BuiltinOperator_ONE_HOT, in RegisterSelectedOps()
246 resolver->AddBuiltin(tflite::BuiltinOperator_POW, in RegisterSelectedOps()
250 resolver->AddBuiltin(tflite::BuiltinOperator_TANH, in RegisterSelectedOps()
254 resolver->AddBuiltin(::tflite::BuiltinOperator_UNIQUE, in RegisterSelectedOps()
256 resolver->AddBuiltin(::tflite::BuiltinOperator_REDUCE_PROD, in RegisterSelectedOps()
258 resolver->AddBuiltin(::tflite::BuiltinOperator_SHAPE, in RegisterSelectedOps()
260 resolver->AddBuiltin(::tflite::BuiltinOperator_NOT_EQUAL, in RegisterSelectedOps()
262 resolver->AddBuiltin(::tflite::BuiltinOperator_CUMSUM, in RegisterSelectedOps()
264 resolver->AddBuiltin(::tflite::BuiltinOperator_EXPAND_DIMS, in RegisterSelectedOps()
266 resolver->AddBuiltin(::tflite::BuiltinOperator_FILL, in RegisterSelectedOps()
268 resolver->AddBuiltin(::tflite::BuiltinOperator_PADV2, in RegisterSelectedOps()
270 resolver->AddBuiltin(::tflite::BuiltinOperator_EMBEDDING_LOOKUP, in RegisterSelectedOps()
274 resolver->AddBuiltin(::tflite::BuiltinOperator_GREATER, in RegisterSelectedOps()
276 resolver->AddBuiltin(::tflite::BuiltinOperator_GELU, in RegisterSelectedOps()
280 void RegisterSelectedOps(tflite::MutableOpResolver* resolver) { in RegisterSelectedOps() argument
281 resolver->AddBuiltin(tflite::BuiltinOperator_FULLY_CONNECTED, in RegisterSelectedOps()
295 std::unique_ptr<tflite::MutableOpResolver> resolver( in BuildOpResolver() local
297 RegisterSelectedOps(resolver.get()); in BuildOpResolver()
299 std::unique_ptr<tflite::ops::builtin::BuiltinOpResolver> resolver( in BuildOpResolver()
303 resolver->AddCustom("DistanceDiversification", in BuildOpResolver()
305 resolver->AddCustom("TextEncoder", in BuildOpResolver()
307 resolver->AddCustom("TextEncoder3S", in BuildOpResolver()
309 resolver->AddCustom("TokenEncoder", in BuildOpResolver()
311 resolver->AddCustom( in BuildOpResolver()
314 resolver->AddCustom("RaggedRange", in BuildOpResolver()
316 resolver->AddCustom( in BuildOpResolver()
319 resolver->AddCustom( in BuildOpResolver()
322 resolver->AddCustom( in BuildOpResolver()
325 resolver->AddCustom("RandomUniform", in BuildOpResolver()
328 customize_fn(resolver.get()); in BuildOpResolver()
329 return std::unique_ptr<tflite::OpResolver>(std::move(resolver)); in BuildOpResolver()
360 std::unique_ptr<tflite::OpResolver> resolver) in TfLiteModelExecutor() argument
361 : model_(std::move(model)), resolver_(std::move(resolver)) {} in TfLiteModelExecutor()