2 * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 #ifndef __LUCI_IR_CIRCLEFULLYCONNECTED_H__
18 #define __LUCI_IR_CIRCLEFULLYCONNECTED_H__
20 #include "luci/IR/CircleNodeDecl.h"
21 #include "luci/IR/CircleOpcode.h"
23 #include "luci/IR/AttrFusedActFunc.h"
24 #include "luci/IR/LuciNodeMixins.h"
30 * @brief FULLY_CONNECTED in Circle
32 class CircleFullyConnected final
33 : public FixedArityNode<3, CircleNodeImpl<CircleOpcode::FULLY_CONNECTED>>,
34 public LuciNodeMixin<LuciNodeTrait::FusedActFunc>,
35 public LuciNodeMixin<LuciNodeTrait::Bias>
38 enum class WeightsFormat
40 UNDEFINED, // This is not defined by Circle. This was added to prevent programming error.
48 loco::Node *input(void) const { return at(0)->node(); }
49 void input(loco::Node *node) { at(0)->node(node); }
51 loco::Node *weights(void) const { return at(1)->node(); }
52 void weights(loco::Node *node) { at(1)->node(node); }
54 loco::Node *bias(void) const override { return at(2)->node(); }
55 void bias(loco::Node *node) override { at(2)->node(node); }
58 WeightsFormat weights_format(void) const { return _weights_format; }
59 void weights_format(WeightsFormat weights_format) { _weights_format = weights_format; }
62 WeightsFormat _weights_format{WeightsFormat::DEFAULT};
67 #endif // __LUCI_IR_CIRCLEFULLYCONNECTED_H__