diff --git a/core/iwasm/libraries/wasi-nn/lib_run_inference.cpp b/core/iwasm/libraries/wasi-nn/lib_run_inference.cpp
index bc4bb0326..b9f13ecef 100644
--- a/core/iwasm/libraries/wasi-nn/lib_run_inference.cpp
+++ b/core/iwasm/libraries/wasi-nn/lib_run_inference.cpp
@@ -43,3 +43,15 @@ _load(graph_builder_array builder, graph_encoding encoding)
 
     return success;
 }
+
+
+uint32_t set_input()
+{
+
+    for (int i=0 ;  i<input_tensor.size()  ;  i++  )
+  	{
+    	input[i]=  input_tensor[i];
+  	}
+
+
+}
diff --git a/core/iwasm/libraries/wasi-nn/wasi_nn.c b/core/iwasm/libraries/wasi-nn/wasi_nn.c
index 7ba70fed9..45c2ed19b 100644
--- a/core/iwasm/libraries/wasi-nn/wasi_nn.c
+++ b/core/iwasm/libraries/wasi-nn/wasi_nn.c
@@ -46,15 +46,16 @@ void
 wasi_nn_init_execution_context()
 {}
 
-void
-wasi_nn_set_input()
+void wasi_nn_set_input(wasm_exec_env_t exec_env ,graph_execution_context context, uint32_t index, tensor tensor)
 {
+    printf("Inside wasi_nn_set_input!\n\n");
+
     // interpreter->AllocateTensors();
 }
 
-void
-wasi_nn_compute()
-{}
+
+void wasi_nn_compute()
+{
 
 void
 wasi_nn_get_output()
diff --git a/core/iwasm/libraries/wasi-nn/wasi_nn.h b/core/iwasm/libraries/wasi-nn/wasi_nn.h
index 184c4d1c2..24a83d8fc 100644
--- a/core/iwasm/libraries/wasi-nn/wasi_nn.h
+++ b/core/iwasm/libraries/wasi-nn/wasi_nn.h
@@ -10,7 +10,14 @@
 
 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;