ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Pool2D.h
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
#ifndef __ONERT_IR_OPERATION_POOL2D_H__
18
#define __ONERT_IR_OPERATION_POOL2D_H__
19
20
#include <memory>
21
22
#include "ir/Operation.h"
23
#include "
ir/InternalType.h
"
24
#include "
ir/Padding.h
"
25
26
namespace
onert
27
{
28
namespace
ir
29
{
30
namespace
operation
31
{
32
33
class
Pool2D
:
public
Operation
34
{
35
public
:
36
enum
Input
37
{
38
INPUT
= 0
39
};
40
41
enum class
PoolType
42
{
43
AVG
,
44
L2
,
45
MAX
,
46
};
47
48
struct
Param
49
{
50
PoolType
op_type
;
51
uint32_t
kh
;
52
uint32_t
kw
;
53
Stride
stride
;
54
Padding
padding
;
55
Activation
activation
;
56
};
57
58
public
:
59
Pool2D
(
const
OperandIndexSequence
&inputs,
const
OperandIndexSequence
&outputs,
60
const
Param
&
param
);
61
62
public
:
63
void
accept
(
OperationVisitor
&v)
const override
;
64
std::string
name
()
const override
;
65
OpCode
opcode
() const final {
return
OpCode::Pool2D; }
66
67
public
:
68
const
Param
&
param
()
const
{
return
_param; }
69
70
private
:
71
Param _param;
72
};
73
74
}
// namespace operation
75
}
// namespace ir
76
}
// namespace onert
77
78
#endif
// __ONERT_IR_OPERATION_POOL2D_H__
InternalType.h
onert::ir::OperandIndexSequence
Definition
OperandIndexSequence.h:37
onert::ir::Operation
Definition
Operation.h:34
onert::ir::operation::Pool2D
Definition
Pool2D.h:34
onert::ir::operation::Pool2D::accept
void accept(OperationVisitor &v) const override
Definition
Pool2D.cc:29
onert::ir::operation::Pool2D::PoolType
PoolType
Definition
Pool2D.h:42
onert::ir::operation::Pool2D::PoolType::MAX
@ MAX
onert::ir::operation::Pool2D::PoolType::L2
@ L2
onert::ir::operation::Pool2D::PoolType::AVG
@ AVG
onert::ir::operation::Pool2D::param
const Param & param() const
Definition
Pool2D.h:68
onert::ir::operation::Pool2D::Input
Input
Definition
Pool2D.h:37
onert::ir::operation::Pool2D::INPUT
@ INPUT
Definition
Pool2D.h:38
onert::ir::operation::Pool2D::name
std::string name() const override
Definition
Pool2D.cc:37
onert::ir::operation::Pool2D::opcode
OpCode opcode() const final
Definition
Pool2D.h:65
onert::ir::OpCode
OpCode
Definition
OpCode.h:30
onert::ir::Activation
Activation
Definition
InternalType.h:28
onert
Definition
CustomKernel.cc:20
Padding.h
onert::ir::OperationVisitor
Definition
OperationVisitor.h:28
onert::ir::Padding
Definition
Padding.h:55
onert::ir::Stride
Definition
InternalType.h:38
onert::ir::operation::Pool2D::Param
Definition
Pool2D.h:49
onert::ir::operation::Pool2D::Param::padding
Padding padding
Definition
Pool2D.h:54
onert::ir::operation::Pool2D::Param::kw
uint32_t kw
Definition
Pool2D.h:52
onert::ir::operation::Pool2D::Param::stride
Stride stride
Definition
Pool2D.h:53
onert::ir::operation::Pool2D::Param::kh
uint32_t kh
Definition
Pool2D.h:51
onert::ir::operation::Pool2D::Param::activation
Activation activation
Definition
Pool2D.h:55
onert::ir::operation::Pool2D::Param::op_type
PoolType op_type
Definition
Pool2D.h:50
runtime
onert
core
include
ir
operation
Pool2D.h
Generated by
1.9.8