ONE - On-device Neural Engine
Loading...
Searching...
No Matches
CircleInstanceNorm.cpp
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2021 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 "
CircleCloneNode.h
"
18
19
namespace
luci
20
{
21
22
luci::CircleNode
*
CloneNode::visit
(
const
luci::CircleInstanceNorm
*node)
23
{
24
if
(node->
fusedActivationFunction
() ==
luci::FusedActFunc::UNDEFINED
)
25
return
nullptr
;
26
27
auto
*cloned =
_graph
->
nodes
()->
create
<
luci::CircleInstanceNorm
>();
28
if
(cloned !=
nullptr
)
29
{
30
cloned->
fusedActivationFunction
(node->
fusedActivationFunction
());
31
cloned->epsilon(node->
epsilon
());
32
}
33
return
cloned;
34
}
35
36
}
// namespace luci
CircleCloneNode.h
loco::Graph::nodes
NodeContext * nodes(void)
Definition
Graph.h:218
loco::NodePool::create
Derived * create(Args &&...args)
Definition
NodePool.h:37
luci::CircleInstanceNorm
INSTANCE_NORM in Circle.
Definition
CircleInstanceNorm.h:35
luci::CircleInstanceNorm::epsilon
float epsilon() const
Definition
CircleInstanceNorm.h:48
luci::CircleNodeMixin< CircleNodeTrait::FusedActFunc >::fusedActivationFunction
FusedActFunc fusedActivationFunction() const
Definition
CircleNodeMixins.h:45
luci::CloneNode::visit
luci::CircleNode * visit(const luci::CircleBCQFullyConnected *) final
Definition
CircleBCQFullyConnected.cpp:22
luci::CloneNode::_graph
loco::Graph * _graph
Definition
CircleCloneNode.h:291
luci
Definition
DepthwiseConv2D.h:24
luci::FusedActFunc::UNDEFINED
@ UNDEFINED
luci::CircleNode
Definition
CircleNodeDecl.h:40
compiler
luci
service
src
Nodes
CircleInstanceNorm.cpp
Generated by
1.9.8