ttnn.MaxPool2d

class ttnn.MaxPool2d(kernel_size: int | Tuple[int, int], stride: int | Tuple[int, int] = 1, padding: int | Tuple[int, int] = 0, dilation: int | Tuple[int, int] = 1, dtype: DataType | None = None, *, device: Device, batch_size: int, input_height: int, input_width: int, reader_patterns_cache: Dict, parallel_config_override: Dict | None = None, deallocate_activation: bool = False, channels: int | None = None)

Applies a 2D max pooling over an input signal composed of several input planes.

If padding is non-zero, then the input is implicitly padded with negative infinity on both sides for padding number of points. dilation controls the spacing between the kernel points.

Arguments:

• attr:

  kernel_size (Union[int, Tuple[int, int]]): the size of the window to take a max over

• attr:

  stride (Union[int, Tuple[int, int]]): the stride of the window. Default value is 1

• attr:

  padding (Union[int, Tuple[int, int]]): Implicit negative infinity padding to be added on both sides

• attr:

  dilation (Union[int, Tuple[int, int]]): a parameter that controls the stride of window elements