yystopf/xiuos
1
0
Fork 1
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

APP_Framework/Framework/:add rt-thread transform layer

Browse Source
This commit is contained in:
chunyexixiaoyu
2021-07-21 13:37:48 +08:00
parent 38d2cb3c85
commit f50dab2bde
4 changed files with 70 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
APP_Framework/Applications/knowing_app/mnist/main.cpp
View File

@@ -1,4 +1,4 @@
#include <rtthread.h>
#include <transform.h>
#include <stdio.h>
#include "tensorflow/lite/micro/all_ops_resolver.h"
@@ -34,7 +34,7 @@ extern "C" void mnist_app() {
return;
}
tensor_arena = (uint8_t *)rt_malloc(kTensorArenaSize);
tensor_arena = (uint8_t *)malloc(kTensorArenaSize);
if (tensor_arena == nullptr) {
TF_LITE_REPORT_ERROR(error_reporter, "malloc for tensor_arena failed");
return;
@@ -90,5 +90,7 @@ extern "C" void mnist_app() {
}
extern "C" {
#ifdef __RT_THREAD_H__
MSH_CMD_EXPORT(mnist_app, run mnist app);
#endif
}