Class ov::op::util::AvgPoolBase#

class AvgPoolBase : public ov::op::Op#

Subclassed by ov::op::v14::AvgPool, ov::op::v1::AvgPool

Public Functions

AvgPoolBase(const Output<Node> &arg, const Strides &strides, const Shape &pads_begin, const Shape &pads_end, const Shape &kernel, bool exclude_pad, RoundingType rounding_type = RoundingType::FLOOR, const PadType &auto_pad = 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.

bool get_exclude_pad() const#
Returns:

Exclude zero-values in padding area.

const PadType &get_auto_pad() const#
Returns:

The pad type for pooling.

RoundingType get_rounding_type() const#
Returns:

The ceiling mode being used for output shape computations