Skip to content

Commit 3e88521

Browse files
committed
Merge branch 'ci/sync_gh_tflite-lib' into 'master'
Update tflite-lib (synced from github) - 538339 See merge request app-frameworks/tflite-micro-esp-examples!98
2 parents 65db24a + c597e9c commit 3e88521

File tree

113 files changed

+675
-422
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

113 files changed

+675
-422
lines changed

components/tflite-lib/tensorflow/lite/micro/kernels/activations.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -109,11 +109,11 @@ TfLiteStatus Relu6Eval(TfLiteContext* context, TfLiteNode* node) {
109109

110110
} // namespace
111111

112-
TfLiteRegistration_V1 Register_RELU() {
112+
TFLMRegistration Register_RELU() {
113113
return tflite::micro::RegisterOp(ReluInit, ReluPrepare, ReluEval);
114114
}
115115

116-
TfLiteRegistration_V1 Register_RELU6() {
116+
TFLMRegistration Register_RELU6() {
117117
return tflite::micro::RegisterOp(Relu6Init, Relu6Prepare, Relu6Eval);
118118
}
119119

components/tflite-lib/tensorflow/lite/micro/kernels/add.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ TfLiteStatus AddEval(TfLiteContext* context, TfLiteNode* node) {
193193
return kTfLiteOk;
194194
}
195195

196-
TfLiteRegistration_V1 Register_ADD() {
196+
TFLMRegistration Register_ADD() {
197197
return tflite::micro::RegisterOp(AddInit, AddPrepare, AddEval);
198198
}
199199

components/tflite-lib/tensorflow/lite/micro/kernels/add.h

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ limitations under the License.
2020

2121
#include "tensorflow/lite/c/builtin_op_data.h"
2222
#include "tensorflow/lite/c/common.h"
23+
#include "tensorflow/lite/micro/micro_common.h"
2324

2425
namespace tflite {
2526

@@ -60,17 +61,17 @@ TfLiteStatus CalculateOpDataAdd(TfLiteContext* context, TfLiteAddParams* params,
6061
TfLiteStatus AddPrepare(TfLiteContext* context, TfLiteNode* node);
6162

6263
// Generic must define registration function.
63-
TfLiteRegistration_V1 Register_ADD();
64+
TFLMRegistration Register_ADD();
6465

6566
#if defined(CMSIS_NN)
66-
TfLiteRegistration_V1 Register_ADD_INT8();
67+
TFLMRegistration Register_ADD_INT8();
6768

68-
TfLiteRegistration_V1 Register_ADD_INT16();
69+
TFLMRegistration Register_ADD_INT16();
6970
#else
7071
// Fallback registration
71-
inline TfLiteRegistration_V1 Register_ADD_INT8() { return Register_ADD(); }
72+
inline TFLMRegistration Register_ADD_INT8() { return Register_ADD(); }
7273

73-
inline TfLiteRegistration_V1 Register_ADD_INT16() { return Register_ADD(); }
74+
inline TFLMRegistration Register_ADD_INT16() { return Register_ADD(); }
7475
#endif
7576
} // namespace tflite
7677

components/tflite-lib/tensorflow/lite/micro/kernels/add_n.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
208208

209209
} // namespace
210210

211-
TfLiteRegistration_V1 Register_ADD_N() {
211+
TFLMRegistration Register_ADD_N() {
212212
return tflite::micro::RegisterOp(nullptr, Prepare, Eval);
213213
}
214214

components/tflite-lib/tensorflow/lite/micro/kernels/arg_min_max.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,11 +107,11 @@ TfLiteStatus ArgMaxEval(TfLiteContext* context, TfLiteNode* node) {
107107

108108
} // namespace
109109

110-
TfLiteRegistration_V1 Register_ARG_MAX() {
110+
TFLMRegistration Register_ARG_MAX() {
111111
return tflite::micro::RegisterOp(nullptr, nullptr, ArgMaxEval);
112112
}
113113

114-
TfLiteRegistration_V1 Register_ARG_MIN() {
114+
TFLMRegistration Register_ARG_MIN() {
115115
return tflite::micro::RegisterOp(nullptr, nullptr, ArgMinEval);
116116
}
117117

components/tflite-lib/tensorflow/lite/micro/kernels/assign_variable.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
100100

101101
} // namespace.
102102

103-
TfLiteRegistration_V1 Register_ASSIGN_VARIABLE() {
103+
TFLMRegistration Register_ASSIGN_VARIABLE() {
104104
return tflite::micro::RegisterOp(nullptr, Prepare, Eval);
105105
}
106106

components/tflite-lib/tensorflow/lite/micro/kernels/batch_to_space_nd.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
105105

106106
} // namespace.
107107

108-
TfLiteRegistration_V1 Register_BATCH_TO_SPACE_ND() {
108+
TFLMRegistration Register_BATCH_TO_SPACE_ND() {
109109
return tflite::micro::RegisterOp(nullptr, Prepare, Eval);
110110
}
111111

components/tflite-lib/tensorflow/lite/micro/kernels/broadcast_args.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ TfLiteStatus BroadcastArgsEval(TfLiteContext* context, TfLiteNode* node) {
8383

8484
} // namespace
8585

86-
TfLiteRegistration_V1 Register_BROADCAST_ARGS() {
86+
TFLMRegistration Register_BROADCAST_ARGS() {
8787
return tflite::micro::RegisterOp(nullptr, BroadcastArgsPrepare,
8888
BroadcastArgsEval);
8989
}

components/tflite-lib/tensorflow/lite/micro/kernels/broadcast_to.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,15 +107,15 @@ TfLiteStatus BroadcastToEval(TfLiteContext* context, TfLiteNode* node) {
107107
micro::GetEvalInput(context, node, kInputTensor);
108108
TfLiteEvalTensor* output = micro::GetEvalOutput(context, node, kOutputTensor);
109109

110-
// BroadcastTo op support upto 5 dims, different from 8 dims in TFLite.
110+
// BroadcastTo op support up to 5 dims, different from 8 dims in TFLite.
111111
reference_ops::BroadcastTo<kMaxDims>(
112112
micro::GetTensorShape(input), input->data.raw,
113113
micro::GetTensorShape(output), output->data.raw, input->type);
114114
return kTfLiteOk;
115115
}
116116
} // namespace
117117

118-
TfLiteRegistration_V1 Register_BROADCAST_TO() {
118+
TFLMRegistration Register_BROADCAST_TO() {
119119
return tflite::micro::RegisterOp(nullptr, BroadcastToPrepare,
120120
BroadcastToEval);
121121
}

components/tflite-lib/tensorflow/lite/micro/kernels/call_once.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
8181

8282
} // namespace.
8383

84-
TfLiteRegistration_V1 Register_CALL_ONCE() {
84+
TFLMRegistration Register_CALL_ONCE() {
8585
return tflite::micro::RegisterOp(Init, Prepare, Eval);
8686
}
8787

0 commit comments

Comments
 (0)