Class MaxPool
Defined in File max_pool.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Derived Type
public mindspore::ops::MaxPoolFusion
(Class MaxPoolFusion)
Class Documentation
-
class MaxPool : public mindspore::ops::BaseOperator
Max pooling operation. Refer to Python API mindspore.ops.MaxPool for more details.
Subclassed by mindspore::ops::MaxPoolFusion
Public Functions
-
inline MaxPool()
Constructor.
-
void Init(const std::vector<int64_t> &kernel_size = {1}, const std::vector<int64_t> &stride = {1}, const PadMode &pad_mode = VALID, const Format &format = NCHW, const std::vector<int64_t> &pad = {0, 0, 0, 0}, const RoundMode &round_mode = FLOOR)
Init. Refer to the parameters of Python API mindspore.ops.MaxPool for the inputs.
-
void set_pad_mode(const PadMode &pad_mode)
Set pad_mode.
-
void set_kernel_size(const std::vector<int64_t> &kernel_size)
Set kernel_size.
-
void set_strides(const std::vector<int64_t> &strides)
Set strides.
-
void set_format(const Format &format)
Set format.
-
void set_pad(const std::vector<int64_t> &pad)
Set pad.
-
void set_round_mode(const RoundMode &round_mode)
Set round_mode.
-
std::vector<int64_t> get_kernel_size() const
Get kernel_size.
- Returns
kernel_size.
-
std::vector<int64_t> get_strides() const
Get strides.
- Returns
strides.
-
PadMode get_pad_mode() const
Get pad_mode.
- Returns
pad_mode.
-
Format get_format() const
Get format.
- Returns
format.
-
std::vector<int64_t> get_pad() const
Get pad.
- Returns
pad.
-
RoundMode get_round_mode() const
Get round_mode.
- Returns
round_mode.
-
inline MaxPool()