ONE - On-device Neural Engine
Loading...
Searching...
No Matches
CircleBCQGather.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::CircleBCQGather
*node)
23
{
24
auto
*cloned =
_graph
->
nodes
()->
create
<
luci::CircleBCQGather
>();
25
if
(cloned !=
nullptr
)
26
{
27
cloned->
axis
(node->
axis
());
28
cloned->input_hidden_size(node->
input_hidden_size
());
29
}
30
return
cloned;
31
}
32
33
}
// 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::CircleBCQGather
BCQ_GATHER in Circle.
Definition
CircleBCQGather.h:32
luci::CircleBCQGather::input_hidden_size
int32_t input_hidden_size(void) const
Definition
CircleBCQGather.h:50
luci::CircleBCQGather::axis
int32_t axis(void) const
Definition
CircleBCQGather.h:47
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
CircleBCQGather.cpp
Generated by
1.9.8