ONE - On-device Neural Engine
Loading...
Searching...
No Matches
CircleSVDF.cpp
Go to the documentation of this file.
1/*
2 * Copyright (c) 2022 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
19namespace
20{
21
22void connect(luci::ConnectNode *cn, const luci::CircleSVDF *node)
23{
24 auto *cloned = loco::must_cast<luci::CircleSVDF *>(cn->find_clone(node));
25
26 luci::CircleNode *input = loco::must_cast<luci::CircleNode *>(node->input());
27 luci::CircleNode *weight_feature = loco::must_cast<luci::CircleNode *>(node->weight_feature());
28 luci::CircleNode *weight_time = loco::must_cast<luci::CircleNode *>(node->weight_time());
29 luci::CircleNode *bias = loco::must_cast<luci::CircleNode *>(node->bias());
30 luci::CircleNode *input_activation_state =
31 loco::must_cast<luci::CircleNode *>(node->input_activation_state());
32
33 cloned->input(cn->find_clone(input));
34 cloned->weight_feature(cn->find_clone(weight_feature));
35 cloned->weight_time(cn->find_clone(weight_time));
36 cloned->bias(cn->find_clone(bias));
37 cloned->input_activation_state(cn->find_clone(input_activation_state));
38}
39
40} // namespace
41
42namespace luci
43{
44
45void ConnectNode::visit(const luci::CircleSVDF *node) { connect(this, node); }
46
47} // namespace luci
SVDF in Circle.
Definition CircleSVDF.h:33
loco::Node * bias(void) const
Definition CircleSVDF.h:47
loco::Node * input_activation_state(void) const
Definition CircleSVDF.h:50
loco::Node * input(void) const
Definition CircleSVDF.h:38
loco::Node * weight_feature(void) const
Definition CircleSVDF.h:41
loco::Node * weight_time(void) const
Definition CircleSVDF.h:44
luci::CircleNode * find_clone(const luci::CircleNode *node)
void visit(const luci::CircleAbs *) final
Definition CircleAbs.cpp:39