Mirror/AI-on-the-edge-device
1
0
Fork 0
mirror of https://github.com/jomjol/AI-on-the-edge-device.git synced 2025-12-13 15:06:53 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update tflite

Browse Source
This commit is contained in:
jomjol
2020-11-08 03:27:52 +01:00
parent 05a0f6fa62
commit 84cea8e3d6
169 changed files with 16367 additions and 11456 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
code/lib/jomjol_tfliteclass/CTfLiteClass.cpp
View File

@@ -152,7 +152,8 @@ bool CTfLiteClass::LoadInputImage(std::string _fn)
void CTfLiteClass::MakeAllocate()
{
static tflite::ops::micro::AllOpsResolver resolver;
// static tflite::ops::micro::AllOpsResolver resolver;
static tflite::AllOpsResolver resolver;
this->interpreter = new tflite::MicroInterpreter(this->model, resolver, this->tensor_arena, this->kTensorArenaSize, this->error_reporter);
TfLiteStatus allocate_status = this->interpreter->AllocateTensors();

5
code/lib/jomjol_tfliteclass/CTfLiteClass.h
View File

@@ -5,7 +5,7 @@
exit(1); \
}
#include "tensorflow/lite/micro/kernels/all_ops_resolver.h"
#include "tensorflow/lite/micro/all_ops_resolver.h"
#include "tensorflow/lite/micro/micro_error_reporter.h"
#include "tensorflow/lite/micro/micro_interpreter.h"
#include "tensorflow/lite/schema/schema_generated.h"
@@ -39,7 +39,8 @@ class CTfLiteClass
const tflite::Model* model;
tflite::MicroInterpreter* interpreter;
TfLiteTensor* output = nullptr;
static tflite::ops::micro::AllOpsResolver *resolver;
// static tflite::ops::micro::AllOpsResolver *resolver;
static tflite::AllOpsResolver resolver;
int kTensorArenaSize;
uint8_t *tensor_arena;