ONE - On-device Neural Engine
Loading...
Searching...
No Matches
CircleRmsNorm.cpp
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2024 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::CircleRmsNorm
*node)
23
{
24
auto
*cloned =
_graph
->
nodes
()->
create
<
luci::CircleRmsNorm
>();
25
if
(cloned !=
nullptr
)
26
{
27
cloned->
epsilon
(node->
epsilon
());
28
}
29
return
cloned;
30
}
31
32
}
// 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::CircleRmsNorm
RMS_NORM in Circle.
Definition
CircleRmsNorm.h:32
luci::CircleRmsNorm::epsilon
float epsilon() const
Definition
CircleRmsNorm.h:41
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::CircleNode
Definition
CircleNodeDecl.h:40
compiler
luci
service
src
Nodes
CircleRmsNorm.cpp
Generated by
1.9.8