/src/WasmEdge/include/host/mock/wasi_nn_module.h
Line | Count | Source (jump to first uncovered line) |
1 | | // SPDX-License-Identifier: Apache-2.0 |
2 | | // SPDX-FileCopyrightText: 2019-2024 Second State INC |
3 | | |
4 | | #pragma once |
5 | | |
6 | | #include "host/mock/wasi_nn_func.h" |
7 | | #include "runtime/instance/module.h" |
8 | | |
9 | | namespace WasmEdge { |
10 | | namespace Host { |
11 | | |
12 | | class WasiNNModuleMock : public Runtime::Instance::ModuleInstance { |
13 | | public: |
14 | 0 | WasiNNModuleMock() : Runtime::Instance::ModuleInstance("wasi_ephemeral_nn") { |
15 | 0 | addHostFunc("load", std::make_unique<WasiNNMock::Load>()); |
16 | 0 | addHostFunc("init_execution_context", |
17 | 0 | std::make_unique<WasiNNMock::InitExecCtx>()); |
18 | 0 | addHostFunc("set_input", std::make_unique<WasiNNMock::SetInput>()); |
19 | 0 | addHostFunc("get_output", std::make_unique<WasiNNMock::GetOutput>()); |
20 | 0 | addHostFunc("compute", std::make_unique<WasiNNMock::Compute>()); |
21 | 0 | } |
22 | | }; |
23 | | |
24 | | } // namespace Host |
25 | | } // namespace WasmEdge |