MkldnnMaxPool3d

Applies max pooling 3D (MKLDNN).

Abstract Signature:

MkldnnMaxPool3d(input: Tensor, kernel_size: int)

PyTorchJAX (Core)Apple MLX

PyTorch

API: torch.mkldnn_max_pool3d

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.lax.reduce_window

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Custom / Partial