ONE - On-device Neural Engine
Loading...
Searching...
No Matches
CircleRank.cpp
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2020 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/Import/Nodes/CircleRank.h
"
18
19
#include <
luci/IR/Nodes/CircleRank.h
>
20
21
#include <
loco.h
>
22
23
namespace
luci
24
{
25
bool
CircleRankGraphBuilder::validate
(
const
ValidateArgs
&args)
const
26
{
27
return
GraphBuilder::validate
(args, 1);
28
}
29
30
CircleNode
*CircleRankGraphBuilder::build_node(
const
circle::OperatorT &,
31
const
std::vector<CircleNode *> &inputs,
32
loco::Graph
*graph)
const
33
{
34
auto
*node = graph->nodes()->create<
CircleRank
>();
35
node->
input
(inputs.at(0));
36
37
return
node;
38
}
39
40
}
// namespace luci
loco::Graph
A neural network graph.
Definition
Graph.h:161
luci::CircleRankGraphBuilder::validate
bool validate(const ValidateArgs &args) const final
Definition
CircleRank.cpp:25
luci::CircleRank
RANK in Circle.
Definition
CircleRank.h:32
luci::CircleRank::input
loco::Node * input(void) const
Definition
CircleRank.h:34
luci::GraphBuilder::validate
bool validate(const ValidateArgs &args, size_t input_cnt) const
Definition
GraphBuilder.h:37
CircleRank.h
CircleRank.h
loco.h
luci
Definition
DepthwiseConv2D.h:24
luci::CircleNode
Definition
CircleNodeDecl.h:40
luci::GraphBuilderBase::ValidateArgs
Definition
GraphBuilderBase.h:35
compiler
luci
import
src
Nodes
CircleRank.cpp
Generated by
1.9.8