ONE - On-device Neural Engine
Loading...
Searching...
No Matches
CirclePad.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/Service/CircleShapeInference.h
"
18
19
#include "
CircleCloneNode.h
"
20
#include "
CircleShapeInferenceHelper.h
"
21
22
namespace
luci
23
{
24
25
luci::CircleNode
*
CloneNodeLet<CN::OPQR>::visit
(
const
luci::CirclePad
*)
26
{
27
return
_graph->nodes()->create<
luci::CirclePad
>();
28
}
29
30
namespace
sinf
31
{
32
33
loco::TensorShape
Algorithm::visit
(
const
luci::CirclePad
*node)
34
{
35
auto
paddings = loco::must_cast<const luci::CircleNode *>(node->
paddings
());
36
auto
circle_input = loco::must_cast<const luci::CircleNode *>(node->
input
());
37
auto
input_shape =
circle_shape
(circle_input);
38
return
pad_shape
(input_shape, paddings);
39
}
40
41
}
// namespace sinf
42
}
// namespace luci
CircleCloneNode.h
CircleShapeInference.h
CircleShapeInferenceHelper.h
loco::TensorShape
Definition
TensorShape.h:29
luci::CirclePad
PAD in Circle.
Definition
CirclePad.h:32
luci::CirclePad::input
loco::Node * input(void) const
Definition
CirclePad.h:34
luci::CirclePad::paddings
loco::Node * paddings(void) const
Definition
CirclePad.h:37
luci::CloneNodeLet
Definition
CircleCloneNode.h:39
luci::sinf::Algorithm::visit
loco::TensorShape visit(const luci::CircleNode *node) final
Default fallback.
Definition
CircleShapeInference.h:42
luci::sinf::pad_shape
loco::TensorShape pad_shape(const loco::TensorShape &input_shape, const luci::CircleNode *paddings)
Definition
CircleShapeInferenceHelper.cpp:164
luci::sinf::circle_shape
loco::TensorShape circle_shape(const luci::CircleNode *node)
Definition
CircleShapeInferenceHelper.cpp:143
luci
Definition
DepthwiseConv2D.h:24
luci::CircleNode
Definition
CircleNodeDecl.h:40
compiler
luci
service
src
Nodes
CirclePad.cpp
Generated by
1.9.8