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