ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Runner.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
3
*
4
* Licensed under the Apache License, Version 2.0 (the "License");
5
* you may not use this file except in compliance with the License.
6
* You may obtain a copy of the License at
7
*
8
* http://www.apache.org/licenses/LICENSE-2.0
9
*
10
* Unless required by applicable law or agreed to in writing, software
11
* distributed under the License is distributed on an "AS IS" BASIS,
12
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
* See the License for the specific language governing permissions and
14
* limitations under the License.
15
*/
16
17
#ifndef __NNKIT_SUPPORT_ONNX_RUNNER_H__
18
#define __NNKIT_SUPPORT_ONNX_RUNNER_H__
19
20
#include "
nnkit/support/onnx/Allocator.h
"
21
#include "
nnkit/support/onnx/TensorSet.h
"
22
23
#include <onnxruntime_c_api.h>
24
25
#include <memory>
26
27
namespace
nnkit
28
{
29
namespace
support
30
{
31
namespace
onnx
32
{
33
34
class
Runner
35
{
36
public
:
37
Runner
(
const
std::string &path);
38
~Runner
(
void
);
39
40
void
prepareInputs
(
void
);
41
void
prepareOutputs
(
void
);
42
43
TensorSet
&
inputs
(
void
) {
return
*_inputs; }
44
TensorSet
&
outputs
(
void
) {
return
*_outputs; }
45
46
void
run
(
void
);
47
48
public
:
49
// Disallow copy
50
Runner
(
const
Runner
&) =
delete
;
51
Runner
&
operator=
(
const
Runner
&) =
delete
;
52
53
private
:
54
OrtEnv *_env;
55
OrtSession *_session;
56
57
std::unique_ptr<Allocator> _allocator;
58
59
std::unique_ptr<TensorSet> _inputs;
60
std::unique_ptr<TensorSet> _outputs;
61
};
62
63
}
// namespace onnx
64
}
// namespace support
65
}
// namespace nnkit
66
67
#endif
// __NNKIT_SUPPORT_ONNX_RUNNER_H__
nnkit::support::onnx::Runner
Definition
Runner.h:35
nnkit::support::onnx::Runner::prepareOutputs
void prepareOutputs(void)
Definition
Runner.cpp:106
nnkit::support::onnx::Runner::operator=
Runner & operator=(const Runner &)=delete
nnkit::support::onnx::Runner::Runner
Runner(const Runner &)=delete
nnkit::support::onnx::Runner::outputs
TensorSet & outputs(void)
Definition
Runner.h:44
nnkit::support::onnx::Runner::prepareInputs
void prepareInputs(void)
Definition
Runner.cpp:54
nnkit::support::onnx::Runner::inputs
TensorSet & inputs(void)
Definition
Runner.h:43
nnkit::support::onnx::Runner::run
void run(void)
Definition
Runner.cpp:158
nnkit::support::onnx::Runner::~Runner
~Runner(void)
Definition
Runner.cpp:41
nnkit::support::onnx::TensorSet
Definition
TensorSet.h:37
Allocator.h
nnkit
Definition
Backend.h:32
TensorSet.h
compiler
nnkit-onnxrt
support
include
nnkit
support
onnx
Runner.h
Generated by
1.9.8