MkldnnMaxPool2d

Applies max pooling 2D (MKLDNN).

Abstract Signature:

MkldnnMaxPool2d(input: Tensor, kernel_size: int, stride: int, padding: int = 0, dilation: int = 1, ceil_mode: bool = False)

PyTorchJAX (Core)KerasApple MLX

PyTorch

API: torch.mkldnn_max_pool2d

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.lax.reduce_window

Strategy: Plugin (max_pool_functional)

Keras

API: keras.ops.max_pool

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.core.max_pool2d

Strategy: Direct Mapping

Official Docs ↗