gnh1201/wasm-micro-runtime
1
0
Fork 0
mirror of https://github.com/bytecodealliance/wasm-micro-runtime.git synced 2025-07-15 08:48:33 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

SP-193 load_function (#4)

Browse Source 
* feat: "load module done"

* fix: "pr fixes"

* fix: "pr fixes"
This commit is contained in:
Ahmedounet 2022-06-13 17:23:32 +02:00 committed by GitHub
parent 521e281ead
commit a601990e13
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 32 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
core/iwasm/libraries/wasi-nn/lib_run_inference.cpp
View File

@ -4,20 +4,38 @@
#include <tensorflow/lite/kernels/register.h>
#include <tensorflow/lite/model.h>
#include <tensorflow/lite/optional_debug_tools.h>
#include <tensorflow/lite/error_reporter.h>
enum Idx {GRAPH=0, GRAPH_SIZE=1 };
std::unique_ptr<tflite::Interpreter> interpreter = NULL;
std::unique_ptr<tflite::FlatBufferModel> model = NULL;
uint32_t _load(graph_builder_array builder, graph_encoding encoding) {
uint32_t *size = (uint32_t*) builder[1];
printf("inside _load: %d\n", *size);
return suces;
// tflite::ErrorReporter *error_reporter;
// model = tflite::FlatBufferModel::BuildFromBuffer(
// (const char *)builder[0],
// 1000, // TODO: find how to pass buffer size
// error_reporter
// );
// tflite::ops::builtin::BuiltinOpResolver resolver;
// tflite::InterpreterBuilder(*model, resolver)(&interpreter);
if(encoding!=tensorflow){return invalid_argument;}
uint32_t *size = (uint32_t*) builder[Idx::GRAPH_SIZE];
tflite::ErrorReporter *error_reporter;
model = tflite::FlatBufferModel::BuildFromBuffer((const char *)builder[Idx::GRAPH], *size, error_reporter);
if(model== nullptr){
printf("failure: null model \n");
return invalid_argument;
}
// Build the interpreter with the InterpreterBuilder.
tflite::ops::builtin::BuiltinOpResolver resolver;
tflite::InterpreterBuilder tflite_builder(*model, resolver);
tflite_builder(&interpreter);
if(interpreter==nullptr){
printf("failure: null interpreter \n");
return invalid_argument;
}
return success;
}

2
core/iwasm/libraries/wasi-nn/wasi_nn.h
View File

@ -11,7 +11,7 @@
typedef uint32_t buffer_size;
typedef enum {
suces = 0,
success = 0,
invalid_argument,
missing_memory,
busy