19 #include "ngraph/op/op.hpp"
20 #include "ngraph/op/util/attr_types.hpp"
32 NGRAPH_RTTI_DECLARATION;
48 const Strides& strides,
49 const Shape& pads_begin,
50 const Shape& pads_end,
65 const Strides& strides,
66 const Shape& pads_begin,
67 const Shape& pads_end,
75 virtual std::shared_ptr<Node>
76 clone_with_new_inputs(
const OutputVector& new_args)
const override;
80 void set_kernel(
const Shape& kernel) { m_kernel = kernel; }
83 void set_strides(
const Strides& strides) { m_strides = strides; }
86 void set_pads_begin(
const Shape& pads_begin) { m_pads_begin = pads_begin; }
89 void set_adding_above(
const Shape& pads_end) { m_pads_end = pads_end; }
92 void set_auto_pad(
const PadType& auto_pad) { m_auto_pad = auto_pad; }
97 m_rounding_type = rounding_mode;
102 bool evaluate(
const HostTensorVector& outputs,
103 const HostTensorVector& inputs)
const override;
114 bool update_auto_padding(
const PartialShape& in_shape,
116 Shape& new_pads_begin)
const;
PadType
Padding Type used for Convolution and Pooling
Definition: attr_types.hpp:71
const Shape & get_kernel() const
Definition: max_pool.hpp:79
const Shape & get_pads_begin() const
Definition: max_pool.hpp:85
op::RoundingType get_rounding_type() const
Definition: max_pool.hpp:94
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
RoundingType
Rounding Type used for Pooling operators.
Definition: attr_types.hpp:101
const Shape & get_pads_end() const
Definition: max_pool.hpp:88
MaxPool()=default
Constructs a batched max pooling operation.
MaxPool(const Output< Node > &arg, const Strides &strides, const Shape &pads_begin, const Shape &pads_end, const Shape &kernel, op::RoundingType rounding_mode, const PadType &auto_pad)
Constructs a batched max pooling operation.
virtual std::shared_ptr< Node > get_default_value() const override
const PadType & get_auto_pad() const
Definition: max_pool.hpp:91
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
MaxPool(const Output< Node > &arg, const Strides &strides, const Shape &pads_begin, const Shape &pads_end, const Shape &kernel, op::RoundingType rounding_mode)
Constructs a batched max pooling operation.
Batched max pooling operation.
Definition: max_pool.hpp:30
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
size_t get_version() const override
Definition: max_pool.hpp:72
const Strides & get_strides() const
Definition: max_pool.hpp:82
Root of all actual ops.
Definition: op.hpp:29