ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Module.cpp
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2020 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
#include "
luci/IR/Module.h
"
18
19
#include <stdexcept>
20
21
namespace
luci
22
{
23
24
void
Module::add
(std::unique_ptr<loco::Graph> &&g)
25
{
26
if
(g.get() ==
nullptr
)
27
throw
std::invalid_argument(
"Module: Graph cannot be null"
);
28
29
_graphs.emplace_back(std::move(g));
30
}
31
32
loco::Graph
*
Module::graph
(
void
)
const
33
{
34
auto
&
graph
= _graphs.at(0);
35
return
graph
.get();
36
}
37
38
loco::Graph
*
Module::graph
(
size_t
idx)
const
39
{
40
auto
&
graph
= _graphs.at(idx);
41
return
graph
.get();
42
}
43
44
std::unique_ptr<Module>
make_module
(
void
) {
return
std::make_unique<Module>(); }
45
46
}
// namespace luci
loco::Graph
A neural network graph.
Definition
Graph.h:161
luci::Module::graph
loco::Graph * graph(void) const
provide main graph
Definition
Module.cpp:32
luci::Module::add
void add(std::unique_ptr< loco::Graph > &&g)
Definition
Module.cpp:24
Module.h
luci
Definition
DepthwiseConv2D.h:24
luci::make_module
std::unique_ptr< Module > make_module(void)
Definition
Module.cpp:44
compiler
luci
lang
src
Module.cpp
Generated by
1.9.8