ONE - On-device Neural Engine
Loading...
Searching...
No Matches
CircleConcatenation.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 "
luci/ConnectNode.h
"
18
19
namespace
20
{
21
22
void
connect(
luci::ConnectNode
*cn,
const
luci::CircleConcatenation
*node)
23
{
24
auto
*cloned = loco::must_cast<luci::CircleConcatenation *>(cn->
find_clone
(node));
25
26
uint32_t
num_inputs
= cloned->numValues();
27
for
(uint32_t i = 0; i <
num_inputs
; ++i)
28
{
29
luci::CircleNode
*value = loco::must_cast<luci::CircleNode *>(node->
values
(i));
30
31
cloned->values(i, cn->
find_clone
(value));
32
}
33
}
34
35
}
// namespace
36
37
namespace
luci
38
{
39
40
void
ConnectNode::visit
(
const
luci::CircleConcatenation
*node) { connect(
this
, node); }
41
42
}
// namespace luci
ConnectNode.h
luci::CircleConcatenation
CONCATENATION in Circle.
Definition
CircleConcatenation.h:38
luci::CircleConcatenation::values
Node * values(uint32_t index) const
Definition
CircleConcatenation.h:51
luci::ConnectNode
Definition
ConnectNode.h:60
luci::ConnectNode::find_clone
luci::CircleNode * find_clone(const luci::CircleNode *node)
Definition
ConnectNode.cpp:30
luci::ConnectNode::visit
void visit(const luci::CircleAbs *) final
Definition
CircleAbs.cpp:39
luci
Definition
DepthwiseConv2D.h:24
part_eval_one.num_inputs
num_inputs
Definition
part_eval_one.py:69
luci::CircleNode
Definition
CircleNodeDecl.h:40
compiler
luci
partition
src
Nodes
CircleConcatenation.cpp
Generated by
1.9.8