Class ov::op::v1::AvgPool

class AvgPool : public ov::op::Op

Batched average pooling operation.

Public Functions

AvgPool() = default

Constructs a batched average pooling operation.

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 = op::RoundingType::FLOOR, const PadType &auto_pad = op::PadType::EXPLICIT)

Constructs a batched average pooling operation.

Parameters

• arg – The output producing the input data batch tensor.[d1, dn]

• strides – The strides.[n]

• pads_begin – The beginning of padding shape.[n]

• pads_end – The end of padding shape.[n]

• kernel – The kernel shape.[n]

• exclude_pad – If false then averages include padding elements, each treated as the number zero. If true, padding elements are entirely ignored when computing averages.

• rounding_type – Whether to use ceiling or floor rounding type while computing output shape.

• auto_pad – Padding type to use for additional padded dimensions

virtual void validate_and_infer_types() override

Verifies that attributes and inputs are consistent and computes output shapes and element types. Must be implemented by concrete child classes so that it can be run any number of times.

Throws if the node is invalid.

const Shape &get_kernel() const

Returns

The kernel shape.

const Strides &get_strides() const

Returns

The strides.

const Shape &get_pads_begin() const

Returns

The beginning of padding shape.

const Shape &get_pads_end() const

Returns

The end of padding shape.

const PadType &get_auto_pad() const

Returns

The pad type for pooling.