19 #include "ngraph/op/op.hpp"
20 #include "ngraph/op/util/attr_types.hpp"
33 NGRAPH_RTTI_DECLARATION;
56 const Strides& strides,
57 const Shape& pads_begin,
58 const Shape& pads_end,
81 const Strides& strides,
82 const Shape& pads_begin,
83 const Shape& pads_end,
92 virtual std::shared_ptr<Node>
93 clone_with_new_inputs(
const OutputVector& new_args)
const override;
97 void set_kernel(
const Shape& kernel);
100 void set_strides(
const Strides& strides);
103 void set_pads_begin(
const Shape& pads_begin);
106 void set_pads_end(
const Shape& pads_end);
107 bool get_exclude_pad()
const;
108 void set_exclude_pad(
bool exclude_pad);
111 void set_auto_pad(
const PadType& auto_pad);
122 bool m_exclude_pad{
true};
123 PadType m_auto_pad{PadType::EXPLICIT};
const Strides & get_strides() const
const Shape & get_pads_begin() const
PadType
Padding Type used for Convolution and Pooling
Definition: attr_types.hpp:71
virtual std::shared_ptr< Node > get_default_value() const override
RoundingType
Rounding Type used for Pooling operators.
Definition: attr_types.hpp:101
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
const PadType & get_auto_pad() const
AvgPool(const Output< Node > &arg, const Strides &strides, const Shape &pads_begin, const Shape &pads_end, const Shape &kernel, bool exclude_pad, op::RoundingType rounding_type)
Constructs a batched average pooling operation.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
AvgPool(const Output< Node > &arg, const Strides &strides, const Shape &pads_begin, const Shape &pads_end, const Shape &kernel, bool exclude_pad, op::RoundingType rounding_type, const PadType &auto_pad)
Constructs a batched average pooling operation.
const Shape & get_kernel() const
size_t get_version() const override
Definition: avg_pool.hpp:88
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
AvgPool()=default
Constructs a batched average pooling operation.
const Shape & get_pads_end() const
Batched average pooling operation.
Definition: avg_pool.hpp:31
Root of all actual ops.
Definition: op.hpp:29