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

feat: "set input"

Browse Source
This commit is contained in:
Ahmedounet 2022-06-13 17:50:26 +02:00
parent bb2c963a34
commit be33067436
3 changed files with 26 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -43,3 +43,15 @@ _load(graph_builder_array builder, graph_encoding encoding)
return success; return success;
} }
uint32_t set_input()
{
for (int i=0 ; i<input_tensor.size() ; i++ )
{
input[i]= input_tensor[i];
}
}

11
core/iwasm/libraries/wasi-nn/wasi_nn.c
View File

@ -46,15 +46,16 @@ void
wasi_nn_init_execution_context() wasi_nn_init_execution_context()
{} {}
void void wasi_nn_set_input(wasm_exec_env_t exec_env ,graph_execution_context context, uint32_t index, tensor tensor)
wasi_nn_set_input()
{ {
printf("Inside wasi_nn_set_input!\n\n");
// interpreter->AllocateTensors(); // interpreter->AllocateTensors();
} }
void
wasi_nn_compute() void wasi_nn_compute()
{} {
void void
wasi_nn_get_output() wasi_nn_get_output()

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

@ -10,7 +10,14 @@
typedef uint32_t buffer_size; typedef uint32_t buffer_size;
typedef enum { success = 0, invalid_argument, missing_memory, busy } nn_erno; typedef uint32_t graph_execution_context;
typedef enum {
success = 0,
invalid_argument,
missing_memory,
busy
} nn_erno;
typedef uint32_t *tensor_dimensions; typedef uint32_t *tensor_dimensions;