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
{
26
cloned->
axis
(node->
axis
());
27
cloned->input_hidden_size(node->
input_hidden_size
());
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::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
CircleModel.h:27
luci::CircleNode
Definition
CircleNodeDecl.h:41
compiler
luci
service
src
Nodes
CircleBCQGather.cpp
Generated by
1.9.8