Home
last modified time | relevance | path

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

1234567

/external/tensorflow/tensorflow/lite/kernels/
Dregister.cc24 TfLiteRegistration* Register_AUDIO_SPECTROGRAM();
25 TfLiteRegistration* Register_MFCC();
26 TfLiteRegistration* Register_DETECTION_POSTPROCESS();
27 TfLiteRegistration* Register_IF();
28 TfLiteRegistration* Register_WHILE();
34 TfLiteRegistration* Register_ABS();
35 TfLiteRegistration* Register_RELU();
36 TfLiteRegistration* Register_RELU_N1_TO_1();
37 TfLiteRegistration* Register_RELU6();
38 TfLiteRegistration* Register_TANH();
[all …]
Dregister_ref.cc24 TfLiteRegistration* Register_AUDIO_SPECTROGRAM();
25 TfLiteRegistration* Register_MFCC();
26 TfLiteRegistration* Register_DETECTION_POSTPROCESS();
34 TfLiteRegistration* Register_ABS();
35 TfLiteRegistration* Register_RELU();
36 TfLiteRegistration* Register_RELU_N1_TO_1();
37 TfLiteRegistration* Register_RELU6();
38 TfLiteRegistration* Register_TANH_REF();
39 TfLiteRegistration* Register_LOGISTIC_REF();
40 TfLiteRegistration* Register_AVERAGE_POOL_REF();
[all …]
Delementwise.cc113 TfLiteRegistration* Register_ABS() { in Register_ABS()
114 static TfLiteRegistration r = { in Register_ABS()
121 TfLiteRegistration* Register_SIN() { in Register_SIN()
122 static TfLiteRegistration r = { in Register_SIN()
129 TfLiteRegistration* Register_COS() { in Register_COS()
130 static TfLiteRegistration r = { in Register_COS()
137 TfLiteRegistration* Register_LOG() { in Register_LOG()
138 static TfLiteRegistration r = { in Register_LOG()
145 TfLiteRegistration* Register_SQRT() { in Register_SQRT()
146 static TfLiteRegistration r = { in Register_SQRT()
[all …]
Dresize_bilinear.cc126 TfLiteRegistration* Register_RESIZE_BILINEAR_REF() { in Register_RESIZE_BILINEAR_REF()
127 static TfLiteRegistration r = { in Register_RESIZE_BILINEAR_REF()
133 TfLiteRegistration* Register_RESIZE_BILINEAR_GENERIC_OPT() { in Register_RESIZE_BILINEAR_GENERIC_OPT()
134 static TfLiteRegistration r = { in Register_RESIZE_BILINEAR_GENERIC_OPT()
140 TfLiteRegistration* Register_RESIZE_BILINEAR_NEON_OPT() { in Register_RESIZE_BILINEAR_NEON_OPT()
141 static TfLiteRegistration r = { in Register_RESIZE_BILINEAR_NEON_OPT()
147 TfLiteRegistration* Register_RESIZE_BILINEAR() { in Register_RESIZE_BILINEAR()
Dresize_nearest_neighbor.cc126 TfLiteRegistration* Register_RESIZE_NEAREST_NEIGHBOR_REF() { in Register_RESIZE_NEAREST_NEIGHBOR_REF()
127 static TfLiteRegistration r = { in Register_RESIZE_NEAREST_NEIGHBOR_REF()
133 TfLiteRegistration* Register_RESIZE_NEAREST_NEIGHBOR_GENERIC_OPT() { in Register_RESIZE_NEAREST_NEIGHBOR_GENERIC_OPT()
134 static TfLiteRegistration r = { in Register_RESIZE_NEAREST_NEIGHBOR_GENERIC_OPT()
141 TfLiteRegistration* Register_RESIZE_NEAREST_NEIGHBOR_NEON_OPT() { in Register_RESIZE_NEAREST_NEIGHBOR_NEON_OPT()
142 static TfLiteRegistration r = { in Register_RESIZE_NEAREST_NEIGHBOR_NEON_OPT()
148 TfLiteRegistration* Register_RESIZE_NEAREST_NEIGHBOR() { in Register_RESIZE_NEAREST_NEIGHBOR()
Dcomparisons.cc328 TfLiteRegistration* Register_EQUAL() { in Register_EQUAL()
329 static TfLiteRegistration r = { in Register_EQUAL()
334 TfLiteRegistration* Register_NOT_EQUAL() { in Register_NOT_EQUAL()
335 static TfLiteRegistration r = {nullptr, nullptr, in Register_NOT_EQUAL()
341 TfLiteRegistration* Register_GREATER() { in Register_GREATER()
342 static TfLiteRegistration r = {nullptr, nullptr, in Register_GREATER()
348 TfLiteRegistration* Register_GREATER_EQUAL() { in Register_GREATER_EQUAL()
349 static TfLiteRegistration r = {nullptr, nullptr, in Register_GREATER_EQUAL()
355 TfLiteRegistration* Register_LESS() { in Register_LESS()
356 static TfLiteRegistration r = { in Register_LESS()
[all …]
Dpooling.cc388 TfLiteRegistration* Register_AVERAGE_POOL_REF() { in Register_AVERAGE_POOL_REF()
389 static TfLiteRegistration r = {pooling::Init, pooling::Free, in Register_AVERAGE_POOL_REF()
395 TfLiteRegistration* Register_MAX_POOL_REF() { in Register_MAX_POOL_REF()
396 static TfLiteRegistration r = {pooling::Init, pooling::Free, in Register_MAX_POOL_REF()
402 TfLiteRegistration* Register_L2_POOL_REF() { in Register_L2_POOL_REF()
403 static TfLiteRegistration r = {pooling::Init, pooling::Free, in Register_L2_POOL_REF()
409 TfLiteRegistration* Register_AVERAGE_POOL_GENERIC_OPT() { in Register_AVERAGE_POOL_GENERIC_OPT()
410 static TfLiteRegistration r = { in Register_AVERAGE_POOL_GENERIC_OPT()
416 TfLiteRegistration* Register_MAX_POOL_GENERIC_OPT() { in Register_MAX_POOL_GENERIC_OPT()
417 static TfLiteRegistration r = {pooling::Init, pooling::Free, in Register_MAX_POOL_GENERIC_OPT()
[all …]
Ddiv.cc151 TfLiteRegistration* Register_DIV_REF() { in Register_DIV_REF()
152 static TfLiteRegistration r = {div::Init, div::Free, div::Prepare, in Register_DIV_REF()
157 TfLiteRegistration* Register_DIV_GENERIC_OPT() { in Register_DIV_GENERIC_OPT()
158 static TfLiteRegistration r = {div::Init, div::Free, div::Prepare, in Register_DIV_GENERIC_OPT()
163 TfLiteRegistration* Register_DIV_NEON_OPT() { in Register_DIV_NEON_OPT()
164 static TfLiteRegistration r = {div::Init, div::Free, div::Prepare, in Register_DIV_NEON_OPT()
169 TfLiteRegistration* Register_DIV() { in Register_DIV()
Dreduce.cc562 TfLiteRegistration* Register_MEAN_REF() { in Register_MEAN_REF()
563 static TfLiteRegistration r = {reduce::Init, reduce::Free, in Register_MEAN_REF()
569 TfLiteRegistration* Register_SUM_REF() { in Register_SUM_REF()
570 static TfLiteRegistration r = {reduce::Init, reduce::Free, in Register_SUM_REF()
575 TfLiteRegistration* Register_REDUCE_PROD_REF() { in Register_REDUCE_PROD_REF()
576 static TfLiteRegistration r = { in Register_REDUCE_PROD_REF()
582 TfLiteRegistration* Register_REDUCE_MAX_REF() { in Register_REDUCE_MAX_REF()
583 static TfLiteRegistration r = { in Register_REDUCE_MAX_REF()
589 TfLiteRegistration* Register_REDUCE_MIN_REF() { in Register_REDUCE_MIN_REF()
590 static TfLiteRegistration r = { in Register_REDUCE_MIN_REF()
[all …]
Dpad.cc276 TfLiteRegistration* Register_PAD_REF() { in Register_PAD_REF()
277 static TfLiteRegistration r = {nullptr, nullptr, pad::Prepare, in Register_PAD_REF()
282 TfLiteRegistration* Register_PAD_GENERIC_OPT() { in Register_PAD_GENERIC_OPT()
283 static TfLiteRegistration r = {nullptr, nullptr, pad::Prepare, in Register_PAD_GENERIC_OPT()
288 TfLiteRegistration* Register_PAD() { return Register_PAD_GENERIC_OPT(); } in Register_PAD()
291 TfLiteRegistration* Register_PADV2_REF() { in Register_PADV2_REF()
292 static TfLiteRegistration r = {nullptr, nullptr, pad::Prepare, in Register_PADV2_REF()
297 TfLiteRegistration* Register_PADV2_GENERIC_OPT() { in Register_PADV2_GENERIC_OPT()
298 static TfLiteRegistration r = {nullptr, nullptr, pad::Prepare, in Register_PADV2_GENERIC_OPT()
303 TfLiteRegistration* Register_PADV2() { return Register_PADV2_GENERIC_OPT(); } in Register_PADV2()
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dregister.cc24 TfLiteRegistration* Register_AUDIO_SPECTROGRAM();
25 TfLiteRegistration* Register_LAYER_NORM_LSTM();
26 TfLiteRegistration* Register_MFCC();
27 TfLiteRegistration* Register_DETECTION_POSTPROCESS();
28 TfLiteRegistration* Register_RELU_1();
34 TfLiteRegistration* Register_RELU();
35 TfLiteRegistration* Register_RELU_N1_TO_1();
36 TfLiteRegistration* Register_RELU6();
37 TfLiteRegistration* Register_TANH();
38 TfLiteRegistration* Register_LOGISTIC();
[all …]
/external/libtextclassifier/utils/
Dtflite-model-executor.cc26 TfLiteRegistration* Register_ADD();
27 TfLiteRegistration* Register_CONCATENATION();
28 TfLiteRegistration* Register_CONV_2D();
29 TfLiteRegistration* Register_FULLY_CONNECTED();
30 TfLiteRegistration* Register_L2_NORMALIZATION();
31 TfLiteRegistration* Register_MUL();
32 TfLiteRegistration* Register_RESHAPE();
33 TfLiteRegistration* Register_SOFTMAX();
34 TfLiteRegistration* Register_GATHER();
35 TfLiteRegistration* Register_TRANSPOSE();
[all …]
/external/tensorflow/tensorflow/lite/
Dmutable_op_resolver_test.cc29 TfLiteRegistration* GetDummyRegistration() { in GetDummyRegistration()
30 static TfLiteRegistration registration = { in GetDummyRegistration()
43 TfLiteRegistration* GetDummy2Registration() { in GetDummy2Registration()
44 static TfLiteRegistration registration = { in GetDummy2Registration()
57 const TfLiteRegistration* found_registration = in TEST()
69 const TfLiteRegistration* found_registration = in TEST()
79 const TfLiteRegistration* found_registration; in TEST()
97 const TfLiteRegistration* found_registration; in TEST()
110 const TfLiteRegistration* found_registration = resolver.FindOp("AWESOME", 1); in TEST()
123 const TfLiteRegistration* found_registration = in TEST()
[all …]
Dmutable_op_resolver.h57 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
59 const TfLiteRegistration* FindOp(const char* op, int version) const override;
61 const TfLiteRegistration* registration, int min_version = 1,
63 void AddCustom(const char* name, const TfLiteRegistration* registration,
71 std::unordered_map<BuiltinOperatorKey, TfLiteRegistration,
74 std::unordered_map<CustomOperatorKey, TfLiteRegistration,
Dmutable_op_resolver.cc20 const TfLiteRegistration* MutableOpResolver::FindOp(tflite::BuiltinOperator op, in FindOp()
26 const TfLiteRegistration* MutableOpResolver::FindOp(const char* op, in FindOp()
33 const TfLiteRegistration* registration, in AddBuiltin()
36 TfLiteRegistration new_registration = *registration; in AddBuiltin()
46 const TfLiteRegistration* registration, in AddCustom()
49 TfLiteRegistration new_registration = *registration; in AddCustom()
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 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op, in FindOp()
63 const TfLiteRegistration* FindOp(const char* op, int version) const override { in FindOp()
68 TfLiteRegistration* constant_return_;
162 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_reg0 = in TEST()
166 const TfLiteRegistration& reg0 = node_and_reg0->second; in TEST()
181 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_reg1 = in TEST()
[all …]
/external/tensorflow/tensorflow/lite/experimental/micro/
Dmicro_mutable_op_resolver.cc20 const TfLiteRegistration* MicroMutableOpResolver::FindOp( in FindOp()
23 const TfLiteRegistration& registration = registrations_[i]; in FindOp()
32 const TfLiteRegistration* MicroMutableOpResolver::FindOp(const char* op, in FindOp()
35 const TfLiteRegistration& registration = registrations_[i]; in FindOp()
46 TfLiteRegistration* registration, in AddBuiltin()
53 TfLiteRegistration* new_registration = &registrations_[registrations_len_]; in AddBuiltin()
63 TfLiteRegistration* registration, in AddCustom()
70 TfLiteRegistration* new_registration = &registrations_[registrations_len_]; in AddCustom()
Dmicro_mutable_op_resolver.h29 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op,
31 const TfLiteRegistration* FindOp(const char* op, int version) const override;
32 void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration,
34 void AddCustom(const char* name, TfLiteRegistration* registration,
38 TfLiteRegistration registrations_[TFLITE_REGISTRATIONS_MAX];
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dall_ops_resolver.cc19 TfLiteRegistration* Register_DEPTHWISE_CONV_2D();
20 TfLiteRegistration* Micro_Register_DEPTHWISE_CONV_2D() { in Micro_Register_DEPTHWISE_CONV_2D()
24 TfLiteRegistration* Register_FULLY_CONNECTED();
25 TfLiteRegistration* Micro_Register_FULLY_CONNECTED() { in Micro_Register_FULLY_CONNECTED()
29 TfLiteRegistration* Register_SOFTMAX();
30 TfLiteRegistration* Micro_Register_SOFTMAX() { return Register_SOFTMAX(); } in Micro_Register_SOFTMAX()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h75 const TfLiteRegistration* registration,
177 std::vector<std::pair<TfLiteNode, TfLiteRegistration>>&
182 const std::vector<std::pair<TfLiteNode, TfLiteRegistration>>&
188 const std::pair<TfLiteNode, TfLiteRegistration>* node_and_registration( in node_and_registration()
304 void* OpInit(const TfLiteRegistration& op_reg, const char* buffer, in OpInit()
311 void OpFree(const TfLiteRegistration& op_reg, void* buffer) { in OpFree()
319 TfLiteStatus OpPrepare(const TfLiteRegistration& op_reg, TfLiteNode* node) { in OpPrepare()
325 TfLiteStatus OpInvoke(const TfLiteRegistration& op_reg, TfLiteNode* node) { in OpInvoke()
382 TfLiteContext* context, TfLiteRegistration registration,
391 TfLiteRegistration registration, const TfLiteIntArray* nodes_to_replace,
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dlogging_op_resolver.cc27 const TfLiteRegistration* base_registration = in LoggingOpResolver()
31 std::unique_ptr<TfLiteRegistration> logging_registation = in LoggingOpResolver()
32 absl::make_unique<TfLiteRegistration>(*base_registration); in LoggingOpResolver()
38 const TfLiteRegistration* LoggingOpResolver::FindOp(BuiltinOperator op, in FindOp()
54 const TfLiteRegistration* LoggingOpResolver::FindOp(const char* op, in FindOp()
Dlogging_op_resolver_test.cc47 TfLiteRegistration conv_registration = {}; in TEST()
53 TfLiteRegistration add_registration = {}; in TEST()
80 TfLiteRegistration conv_registration = {}; in TEST()
86 TfLiteRegistration add_registration = {}; in TEST()
106 TfLiteRegistration conv_registration = {}; in TEST()
112 TfLiteRegistration add_registration = {}; in TEST()
/external/tensorflow/tensorflow/lite/core/api/
Dop_resolver_test.cc43 const TfLiteRegistration* FindOp(BuiltinOperator op, in FindOp()
46 static TfLiteRegistration r = {MockInit, MockFree, MockPrepare, in FindOp()
53 const TfLiteRegistration* FindOp(const char* op, int version) const override { in FindOp()
55 static TfLiteRegistration r = {MockInit, MockFree, MockPrepare, in FindOp()
86 const TfLiteRegistration* registration = in TEST()
119 const TfLiteRegistration* registration = nullptr; in TEST()
142 const TfLiteRegistration* registration = nullptr; in TEST()
162 const TfLiteRegistration* registration = nullptr; in TEST()
185 const TfLiteRegistration* registration = nullptr; in TEST()
/external/tensorflow/tensorflow/lite/tools/
Dverifier.h29 const TfLiteRegistration* FindOp(tflite::BuiltinOperator op, in FindOp()
31 static TfLiteRegistration null_registration = {nullptr, nullptr, nullptr, in FindOp()
35 const TfLiteRegistration* FindOp(const char* op, int version) const override { in FindOp()
36 static TfLiteRegistration null_registration = {nullptr, nullptr, nullptr, in FindOp()
/external/tensorflow/tensorflow/lite/profiling/
Dprofile_summarizer_test.cc53 TfLiteRegistration* RegisterSimpleOp() { in RegisterSimpleOp()
54 static TfLiteRegistration registration = { in RegisterSimpleOp()
61 TfLiteRegistration* RegisterSimpleOpWithProfilingDetails() { in RegisterSimpleOpWithProfilingDetails()
62 static TfLiteRegistration registration = {nullptr, in RegisterSimpleOpWithProfilingDetails()
76 void Init(const std::function<TfLiteRegistration*()>& registration);
90 const std::function<TfLiteRegistration*()>& registration) { in Init()

1234567