ONE - On-device Neural Engine
Loading...
Searching...
No Matches
CircleRelu.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/Import/Nodes/CircleRelu.h
"
18
19
#include <
luci/IR/Nodes/CircleRelu.h
>
20
21
#include <
loco.h
>
22
23
namespace
luci
24
{
25
26
bool
CircleReluGraphBuilder::validate
(
const
ValidateArgs
&args)
const
27
{
28
return
GraphBuilder::validate
(args, 1);
29
}
30
31
CircleNode
*CircleReluGraphBuilder::build_node(
const
circle::OperatorT &,
32
const
std::vector<CircleNode *> &inputs,
33
loco::Graph
*graph)
const
34
{
35
auto
*node = graph->nodes()->create<
CircleRelu
>();
36
node->
features
(inputs.at(0));
37
38
return
node;
39
}
40
41
}
// namespace luci
loco::Graph
A neural network graph.
Definition
Graph.h:161
luci::CircleReluGraphBuilder::validate
bool validate(const ValidateArgs &args) const final
Definition
CircleRelu.cpp:26
luci::CircleRelu
RELU in Circle.
Definition
CircleRelu.h:32
luci::CircleRelu::features
loco::Node * features(void) const
Definition
CircleRelu.h:34
luci::GraphBuilder::validate
bool validate(const ValidateArgs &args, size_t input_cnt) const
Definition
GraphBuilder.h:37
CircleRelu.h
CircleRelu.h
loco.h
luci
Definition
DepthwiseConv2D.h:24
luci::CircleNode
Definition
CircleNodeDecl.h:40
luci::GraphBuilderBase::ValidateArgs
Definition
GraphBuilderBase.h:35
compiler
luci
import
src
Nodes
CircleRelu.cpp
Generated by
1.9.8