Home
last modified time | relevance | path

Searched refs:TfLiteRegistration (Results 1 – 25 of 63) sorted by relevance

123

/external/tensorflow/tensorflow/contrib/lite/kernels/
Dregister.cc22 TfLiteRegistration* Register_RELU();
23 TfLiteRegistration* Register_RELU_N1_TO_1();
24 TfLiteRegistration* Register_RELU6();
25 TfLiteRegistration* Register_TANH();
26 TfLiteRegistration* Register_LOGISTIC();
27 TfLiteRegistration* Register_AVERAGE_POOL_2D();
28 TfLiteRegistration* Register_MAX_POOL_2D();
29 TfLiteRegistration* Register_L2_POOL_2D();
30 TfLiteRegistration* Register_CONV_2D();
31 TfLiteRegistration* Register_DEPTHWISE_CONV_2D();
[all …]
Dregister.h29 TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const override;
30 TfLiteRegistration* FindOp(const char* op) const override;
31 void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration);
32 void AddCustom(const char* name, TfLiteRegistration* registration);
40 std::unordered_map<tflite::BuiltinOperator, TfLiteRegistration*,
43 std::unordered_map<std::string, TfLiteRegistration*> custom_ops_;
Dpooling.cc299 TfLiteRegistration* Register_AVERAGE_POOL_REF() { in Register_AVERAGE_POOL_REF()
300 static TfLiteRegistration r = {pooling::Init, pooling::Free, in Register_AVERAGE_POOL_REF()
306 TfLiteRegistration* Register_MAX_POOL_REF() { in Register_MAX_POOL_REF()
307 static TfLiteRegistration r = {pooling::Init, pooling::Free, in Register_MAX_POOL_REF()
313 TfLiteRegistration* Register_L2_POOL_REF() { in Register_L2_POOL_REF()
314 static TfLiteRegistration r = {pooling::Init, pooling::Free, in Register_L2_POOL_REF()
320 TfLiteRegistration* Register_AVERAGE_POOL_GENERIC_OPT() { in Register_AVERAGE_POOL_GENERIC_OPT()
321 static TfLiteRegistration r = { in Register_AVERAGE_POOL_GENERIC_OPT()
327 TfLiteRegistration* Register_MAX_POOL_GENERIC_OPT() { in Register_MAX_POOL_GENERIC_OPT()
328 static TfLiteRegistration r = {pooling::Init, pooling::Free, in Register_MAX_POOL_GENERIC_OPT()
[all …]
Dsub.cc101 TfLiteRegistration* Register_SUB_REF() { in Register_SUB_REF()
102 static TfLiteRegistration r = {nullptr, nullptr, sub::Prepare, in Register_SUB_REF()
107 TfLiteRegistration* Register_SUB_GENERIC_OPT() { in Register_SUB_GENERIC_OPT()
108 static TfLiteRegistration r = {nullptr, nullptr, sub::Prepare, in Register_SUB_GENERIC_OPT()
113 TfLiteRegistration* Register_SUB_NEON_OPT() { in Register_SUB_NEON_OPT()
114 static TfLiteRegistration r = {nullptr, nullptr, sub::Prepare, in Register_SUB_NEON_OPT()
119 TfLiteRegistration* Register_SUB() { in Register_SUB()
Ddiv.cc101 TfLiteRegistration* Register_DIV_REF() { in Register_DIV_REF()
102 static TfLiteRegistration r = {nullptr, nullptr, div::Prepare, in Register_DIV_REF()
107 TfLiteRegistration* Register_DIV_GENERIC_OPT() { in Register_DIV_GENERIC_OPT()
108 static TfLiteRegistration r = {nullptr, nullptr, div::Prepare, in Register_DIV_GENERIC_OPT()
113 TfLiteRegistration* Register_DIV_NEON_OPT() { in Register_DIV_NEON_OPT()
114 static TfLiteRegistration r = {nullptr, nullptr, div::Prepare, in Register_DIV_NEON_OPT()
119 TfLiteRegistration* Register_DIV() { in Register_DIV()
Dresize_bilinear.cc114 TfLiteRegistration* Register_RESIZE_BILINEAR_REF() { in Register_RESIZE_BILINEAR_REF()
115 static TfLiteRegistration r = { in Register_RESIZE_BILINEAR_REF()
121 TfLiteRegistration* Register_RESIZE_BILINEAR_GENERIC_OPT() { in Register_RESIZE_BILINEAR_GENERIC_OPT()
122 static TfLiteRegistration r = { in Register_RESIZE_BILINEAR_GENERIC_OPT()
128 TfLiteRegistration* Register_RESIZE_BILINEAR_NEON_OPT() { in Register_RESIZE_BILINEAR_NEON_OPT()
129 static TfLiteRegistration r = { in Register_RESIZE_BILINEAR_NEON_OPT()
135 TfLiteRegistration* Register_RESIZE_BILINEAR() { in Register_RESIZE_BILINEAR()
Dtest_util.h90 SingleOpResolver(const BuiltinOperator op, TfLiteRegistration* registration) in SingleOpResolver()
92 TfLiteRegistration* FindOp(BuiltinOperator op) const override { in FindOp()
98 TfLiteRegistration* FindOp(const char* op) const override { return nullptr; } in FindOp()
102 TfLiteRegistration* registration_;
146 const std::function<TfLiteRegistration*()>& registeration);
219 std::map<string, std::function<TfLiteRegistration*()>> custom_registrations_;
237 const std::map<string, TfLiteRegistration*>& kernel_map) { in GetKernelTags()
246 virtual const std::map<string, TfLiteRegistration*>& GetKernelMap() = 0;
247 TfLiteRegistration* GetRegistration() { in GetRegistration()
Dactivations.cc342 TfLiteRegistration* Register_RELU() { in Register_RELU()
343 static TfLiteRegistration r = {/*init=*/nullptr, /*free=*/nullptr, in Register_RELU()
349 TfLiteRegistration* Register_RELU_N1_TO_1() { in Register_RELU_N1_TO_1()
350 static TfLiteRegistration r = {/*init=*/nullptr, /*free=*/nullptr, in Register_RELU_N1_TO_1()
356 TfLiteRegistration* Register_RELU6() { in Register_RELU6()
357 static TfLiteRegistration r = {/*init=*/nullptr, /*free=*/nullptr, in Register_RELU6()
363 TfLiteRegistration* Register_TANH() { in Register_TANH()
364 static TfLiteRegistration r = {/*init=*/nullptr, /*free=*/nullptr, in Register_TANH()
370 TfLiteRegistration* Register_LOGISTIC() { in Register_LOGISTIC()
371 static TfLiteRegistration r = {activations::Init, activations::Free, in Register_LOGISTIC()
[all …]
Dl2norm.cc88 TfLiteRegistration* Register_L2NORM_REF() { in Register_L2NORM_REF()
89 static TfLiteRegistration r = {nullptr, nullptr, l2norm::Prepare, in Register_L2NORM_REF()
94 TfLiteRegistration* Register_L2NORM_GENERIC_OPT() { in Register_L2NORM_GENERIC_OPT()
95 static TfLiteRegistration r = {nullptr, nullptr, l2norm::Prepare, in Register_L2NORM_GENERIC_OPT()
100 TfLiteRegistration* Register_L2_NORMALIZATION() { in Register_L2_NORMALIZATION()
Dmul.cc172 TfLiteRegistration* Register_MUL_REF() { in Register_MUL_REF()
173 static TfLiteRegistration r = {mul::Init, mul::Free, mul::Prepare, in Register_MUL_REF()
178 TfLiteRegistration* Register_MUL_GENERIC_OPT() { in Register_MUL_GENERIC_OPT()
179 static TfLiteRegistration r = {mul::Init, mul::Free, mul::Prepare, in Register_MUL_GENERIC_OPT()
184 TfLiteRegistration* Register_MUL_NEON_OPT() { in Register_MUL_NEON_OPT()
185 static TfLiteRegistration r = {mul::Init, mul::Free, mul::Prepare, in Register_MUL_NEON_OPT()
190 TfLiteRegistration* Register_MUL() { in Register_MUL()
Dlocal_response_norm.cc89 TfLiteRegistration* Register_LOCAL_RESPONSE_NORM_REF() { in Register_LOCAL_RESPONSE_NORM_REF()
90 static TfLiteRegistration r = { in Register_LOCAL_RESPONSE_NORM_REF()
96 TfLiteRegistration* Register_LOCAL_RESPONSE_NORM_GENERIC_OPT() { in Register_LOCAL_RESPONSE_NORM_GENERIC_OPT()
97 static TfLiteRegistration r = { in Register_LOCAL_RESPONSE_NORM_GENERIC_OPT()
103 TfLiteRegistration* Register_LOCAL_RESPONSE_NORMALIZATION() { in Register_LOCAL_RESPONSE_NORMALIZATION()
Dfully_connected.cc264 TfLiteRegistration* Register_FULLY_CONNECTED_REF() { in Register_FULLY_CONNECTED_REF()
265 static TfLiteRegistration r = { in Register_FULLY_CONNECTED_REF()
271 TfLiteRegistration* Register_FULLY_CONNECTED_NEON_OPT() { in Register_FULLY_CONNECTED_NEON_OPT()
272 static TfLiteRegistration r = { in Register_FULLY_CONNECTED_NEON_OPT()
278 TfLiteRegistration* Register_FULLY_CONNECTED_GENERIC_OPT() { in Register_FULLY_CONNECTED_GENERIC_OPT()
279 static TfLiteRegistration r = { in Register_FULLY_CONNECTED_GENERIC_OPT()
285 TfLiteRegistration* Register_FULLY_CONNECTED_PIE() { in Register_FULLY_CONNECTED_PIE()
286 static TfLiteRegistration r = {fully_connected::Init, fully_connected::Free, in Register_FULLY_CONNECTED_PIE()
292 TfLiteRegistration* Register_FULLY_CONNECTED() { in Register_FULLY_CONNECTED()
Dadd.cc188 TfLiteRegistration* Register_ADD_REF() { in Register_ADD_REF()
189 static TfLiteRegistration r = {add::Init, add::Free, add::Prepare, in Register_ADD_REF()
194 TfLiteRegistration* Register_ADD_GENERIC_OPT() { in Register_ADD_GENERIC_OPT()
195 static TfLiteRegistration r = {add::Init, add::Free, add::Prepare, in Register_ADD_GENERIC_OPT()
200 TfLiteRegistration* Register_ADD_NEON_OPT() { in Register_ADD_NEON_OPT()
201 static TfLiteRegistration r = {add::Init, add::Free, add::Prepare, in Register_ADD_NEON_OPT()
206 TfLiteRegistration* Register_ADD() { in Register_ADD()
Dspace_to_depth.cc126 TfLiteRegistration* Register_SPACE_TO_DEPTH_REF() { in Register_SPACE_TO_DEPTH_REF()
127 static TfLiteRegistration r = { in Register_SPACE_TO_DEPTH_REF()
133 TfLiteRegistration* Register_SPACE_TO_DEPTH_GENERIC_OPT() { in Register_SPACE_TO_DEPTH_GENERIC_OPT()
134 static TfLiteRegistration r = { in Register_SPACE_TO_DEPTH_GENERIC_OPT()
140 TfLiteRegistration* Register_SPACE_TO_DEPTH() { in Register_SPACE_TO_DEPTH()
Dconv_test.cc29 TfLiteRegistration* Register_CONVOLUTION_REF();
30 TfLiteRegistration* Register_CONVOLUTION_GENERIC_OPT();
31 TfLiteRegistration* Register_CONVOLUTION_MULTITHREADED_OPT();
32 TfLiteRegistration* Register_CONVOLUTION_CBLAS_OPT();
46 TfLiteRegistration* registration, const TensorData& input, in BaseConvolutionOpModel()
104 const auto kKernelMap = new std::map<string, TfLiteRegistration*>({
114 const std::map<string, TfLiteRegistration*>& GetKernelMap() override { in GetKernelMap()
Dpad.cc167 TfLiteRegistration* Register_PAD_REF() { in Register_PAD_REF()
168 static TfLiteRegistration r = {nullptr, nullptr, pad::Prepare, in Register_PAD_REF()
173 TfLiteRegistration* Register_PAD_GENERIC_OPT() { in Register_PAD_GENERIC_OPT()
174 static TfLiteRegistration r = {nullptr, nullptr, pad::Prepare, in Register_PAD_GENERIC_OPT()
179 TfLiteRegistration* Register_PAD() { return Register_PAD_GENERIC_OPT(); } in Register_PAD()
Dspace_to_batch_nd.cc165 TfLiteRegistration* Register_SPACE_TO_BATCH_ND_REF() { in Register_SPACE_TO_BATCH_ND_REF()
166 static TfLiteRegistration r = { in Register_SPACE_TO_BATCH_ND_REF()
172 TfLiteRegistration* Register_SPACE_TO_BATCH_ND_GENERIC_OPT() { in Register_SPACE_TO_BATCH_ND_GENERIC_OPT()
173 static TfLiteRegistration r = { in Register_SPACE_TO_BATCH_ND_GENERIC_OPT()
179 TfLiteRegistration* Register_SPACE_TO_BATCH_ND() { in Register_SPACE_TO_BATCH_ND()
Dbatch_to_space_nd.cc168 TfLiteRegistration* Register_BATCH_TO_SPACE_ND_REF() { in Register_BATCH_TO_SPACE_ND_REF()
169 static TfLiteRegistration r = { in Register_BATCH_TO_SPACE_ND_REF()
175 TfLiteRegistration* Register_BATCH_TO_SPACE_ND_GENERIC_OPT() { in Register_BATCH_TO_SPACE_ND_GENERIC_OPT()
176 static TfLiteRegistration r = { in Register_BATCH_TO_SPACE_ND_GENERIC_OPT()
182 TfLiteRegistration* Register_BATCH_TO_SPACE_ND() { in Register_BATCH_TO_SPACE_ND()
Dconcatenation.cc148 TfLiteRegistration* Register_CONCATENATION_REF() { in Register_CONCATENATION_REF()
149 static TfLiteRegistration r = { in Register_CONCATENATION_REF()
155 TfLiteRegistration* Register_CONCATENATION_GENERIC_OPT() { in Register_CONCATENATION_GENERIC_OPT()
156 static TfLiteRegistration r = { in Register_CONCATENATION_GENERIC_OPT()
162 TfLiteRegistration* Register_CONCATENATION() { in Register_CONCATENATION()
Ddepthwise_conv.cc258 TfLiteRegistration* Register_DEPTHWISE_CONVOLUTION_REF() { in Register_DEPTHWISE_CONVOLUTION_REF()
259 static TfLiteRegistration r = { in Register_DEPTHWISE_CONVOLUTION_REF()
265 TfLiteRegistration* Register_DEPTHWISE_CONVOLUTION_GENERIC_OPT() { in Register_DEPTHWISE_CONVOLUTION_GENERIC_OPT()
266 static TfLiteRegistration r = { in Register_DEPTHWISE_CONVOLUTION_GENERIC_OPT()
272 TfLiteRegistration* Register_DEPTHWISE_CONVOLUTION_NEON_OPT() { in Register_DEPTHWISE_CONVOLUTION_NEON_OPT()
273 static TfLiteRegistration r = { in Register_DEPTHWISE_CONVOLUTION_NEON_OPT()
279 TfLiteRegistration* Register_DEPTHWISE_CONV_2D() { in Register_DEPTHWISE_CONV_2D()
/external/tensorflow/tensorflow/contrib/lite/tools/
Dmutable_op_resolver.h43 TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const override;
44 TfLiteRegistration* FindOp(const char* op) const override;
45 void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration);
46 void AddCustom(const char* name, TfLiteRegistration* registration);
49 std::map<int, TfLiteRegistration*> builtins_;
50 std::map<std::string, TfLiteRegistration*> custom_ops_;
Dmutable_op_resolver.cc20 TfLiteRegistration* MutableOpResolver::FindOp( in FindOp()
26 TfLiteRegistration* MutableOpResolver::FindOp(const char* op) const { in FindOp()
32 TfLiteRegistration* registration) { in AddBuiltin()
38 TfLiteRegistration* registration) { in AddCustom()
/external/tensorflow/tensorflow/contrib/lite/
Dinterpreter.h123 const TfLiteRegistration* registration,
194 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_registration( in node_and_registration()
264 void* OpInit(const TfLiteRegistration& op_reg, const char* buffer, in OpInit()
271 void OpFree(const TfLiteRegistration& op_reg, void* buffer) { in OpFree()
279 TfLiteStatus OpPrepare(const TfLiteRegistration& op_reg, TfLiteNode* node) { in OpPrepare()
285 TfLiteStatus OpInvoke(const TfLiteRegistration& op_reg, TfLiteNode* node) { in OpInvoke()
342 TfLiteContext* context, TfLiteRegistration registration,
350 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace);
355 TfLiteRegistration** registration);
361 TfLiteRegistration** registration);
[all …]
Dmodel_test.cc33 inline bool operator==(const TfLiteRegistration& a, in operator ==()
34 const TfLiteRegistration& b) { in operator ==()
47 TfLiteRegistration dummy_reg = {dummy_init, dummy_free, dummy_resize,
55 explicit TrivialResolver(TfLiteRegistration* constant_return = nullptr) in TrivialResolver()
58 TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const override { in FindOp()
62 TfLiteRegistration* FindOp(const char* op) const override { in FindOp()
67 TfLiteRegistration* constant_return_;
160 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_reg0 = in TEST()
164 const TfLiteRegistration& reg0 = node_and_reg0->second; in TEST()
179 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_reg1 = in TEST()
[all …]
Dcontext.h42 typedef struct _TfLiteRegistration TfLiteRegistration; typedef
284 TfLiteRegistration** registration);
288 struct TfLiteContext*, TfLiteRegistration registration,
337 } TfLiteRegistration; typedef

123