diff --git a/src/runtime/crt/aot_executor_module/aot_executor_module.c b/src/runtime/crt/aot_executor_module/aot_executor_module.c index a5c8105144f79..3e42409c53b63 100644 --- a/src/runtime/crt/aot_executor_module/aot_executor_module.c +++ b/src/runtime/crt/aot_executor_module/aot_executor_module.c @@ -210,6 +210,8 @@ static const TVMBackendPackedCFunc aot_executor_registry_funcs[] = { &TVMAotExecutorModule_NotImplemented, // set_input (implemented via python wrapper) &TVMAotExecutorModule_NotImplemented, // share_params (do not implement) &TVMAotExecutorModule_GetInputName, // get_input_name + &TVMAotExecutorModule_NotImplemented, // get_output_index + &TVMAotExecutorModule_NotImplemented, // get_output_info }; static const TVMFuncRegistry aot_executor_registry = { @@ -223,7 +225,9 @@ static const TVMFuncRegistry aot_executor_registry = { "run\0" "set_input\0" "share_params\0" - "get_input_name\0", + "get_input_name\0" + "get_output_index\0" + "get_output_info\0", aot_executor_registry_funcs}; tvm_crt_error_t TVMAotExecutorModule_Register() { diff --git a/src/runtime/crt/graph_executor_module/graph_executor_module.c b/src/runtime/crt/graph_executor_module/graph_executor_module.c index 559b6896a55e9..4a465f60dc66e 100644 --- a/src/runtime/crt/graph_executor_module/graph_executor_module.c +++ b/src/runtime/crt/graph_executor_module/graph_executor_module.c @@ -235,6 +235,8 @@ static const TVMBackendPackedCFunc graph_executor_registry_funcs[] = { &TVMGraphExecutorModule_Run, &TVMGraphExecutorModule_SetInput, &TVMGraphExecutorModule_NotImplemented, // share_params + &TVMGraphExecutorModule_NotImplemented, // get_output_index + &TVMGraphExecutorModule_NotImplemented, // get_output_info }; static const TVMFuncRegistry graph_executor_registry = { @@ -247,7 +249,9 @@ static const TVMFuncRegistry graph_executor_registry = { "load_params\0" "run\0" "set_input\0" - "share_params\0", + "share_params\0" + "get_output_index\0" + "get_output_info\0", graph_executor_registry_funcs}; tvm_crt_error_t TVMGraphExecutorModule_Register() {