ONE - On-device Neural Engine
Loading...
Searching...
No Matches
CircleEqual.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/CircleEqual.h
"
18
19
#include <
luci/IR/Nodes/CircleEqual.h
>
20
21
#include <
loco.h
>
22
23
namespace
luci
24
{
25
26
bool
CircleEqualGraphBuilder::validate
(
const
ValidateArgs
&args)
const
27
{
28
if
(!
GraphBuilder::validate
(args, 2))
29
return
false
;
30
31
const
auto
&inputs = args.op.inputs;
32
const
auto
tensors = args.reader.tensors();
33
34
assert(tensors[inputs.at(0)] !=
nullptr
&& tensors[inputs.at(1)] !=
nullptr
);
35
return
tensors[inputs.at(0)]->type() == tensors[inputs.at(1)]->type();
36
}
37
38
CircleNode
*CircleEqualGraphBuilder::build_node(
const
circle::OperatorT &,
39
const
std::vector<CircleNode *> &inputs,
40
loco::Graph
*graph)
const
41
{
42
auto
*node = graph->nodes()->create<
CircleEqual
>();
43
node->
x
(inputs.at(0));
44
node->y(inputs.at(1));
45
46
return
node;
47
}
48
49
}
// namespace luci
loco::Graph
A neural network graph.
Definition
Graph.h:161
luci::CircleEqualGraphBuilder::validate
bool validate(const ValidateArgs &args) const final
Definition
CircleEqual.cpp:26
luci::CircleEqual
EQUAL in Circle.
Definition
CircleEqual.h:32
luci::CircleEqual::x
loco::Node * x(void) const
Definition
CircleEqual.h:34
luci::GraphBuilder::validate
bool validate(const ValidateArgs &args, size_t input_cnt) const
Definition
GraphBuilder.h:37
CircleEqual.h
CircleEqual.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
CircleEqual.cpp
Generated by
1.9.8