Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -3,15 +3,26 @@
* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
*/

#ifndef WASI_NN_OPENVINO_HPP
#define WASI_NN_OPENVINO_HPP
#ifndef WASI_NN_BACKEND_H
#define WASI_NN_BACKEND_H

#include "wasi_nn_types.h"

#ifdef __cplusplus
extern "C" {
#endif

__attribute__((visibility("default"))) wasi_nn_error
load(void *ctx, graph_builder_array *builder, graph_encoding encoding,
execution_target target, graph *g);

__attribute__((visibility("default"))) wasi_nn_error
load_by_name(void *tflite_ctx, const char *name, uint32_t namelen, graph *g);

__attribute__((visibility("default"))) wasi_nn_error
load_by_name_with_config(void *ctx, const char *name, uint32_t namelen,
const char *config, uint32_t config_len, graph *g);

__attribute__((visibility("default"))) wasi_nn_error
init_execution_context(void *ctx, graph g, graph_execution_context *exec_ctx);

Expand All @@ -32,4 +43,8 @@ init_backend(void **ctx);
__attribute__((visibility("default"))) wasi_nn_error
deinit_backend(void *ctx);

#endif /* WASI_NN_OPENVINO_HPP */
#ifdef __cplusplus
}
#endif

#endif /* WASI_NN_BACKEND_H */
2 changes: 1 addition & 1 deletion core/iwasm/libraries/wasi-nn/src/wasi_nn_llamacpp.c
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

#include <stdlib.h>

#include "wasi_nn_types.h"
#include "wasi_nn_backend.h"
#include "utils/logger.h"
#include "llama.h"
#include "ggml.h"
Expand Down
3 changes: 1 addition & 2 deletions core/iwasm/libraries/wasi-nn/src/wasi_nn_openvino.c
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,7 @@
* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
*/

#include "wasi_nn_types.h"
#include "wasi_nn_openvino.h"
#include "wasi_nn_backend.h"
#include "utils/logger.h"
#include "bh_platform.h"

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,10 @@
* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
*/

#include "wasi_nn_tensorflowlite.hpp"
#include "utils/logger.h"

#include "bh_platform.h"
#include "wasi_nn_types.h"
#include "wasi_nn_backend.h"
#include "wasm_export.h"

#include <tensorflow/lite/interpreter.h>
Expand Down
47 changes: 0 additions & 47 deletions core/iwasm/libraries/wasi-nn/src/wasi_nn_tensorflowlite.hpp

This file was deleted.

Loading