MkldnnAdaptiveAvgPool2d

Applies adaptive average pooling (MKLDNN backend optimization).

Abstract Signature:

MkldnnAdaptiveAvgPool2d(input: Tensor, output_size: int)

PyTorchJAX (Core)KerasApple MLX

PyTorch

API: torch.mkldnn_adaptive_avg_pool2d

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.image.resize

Strategy: Plugin (adaptive_pool)

Keras

API: keras.layers.GlobalAveragePooling2D

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.AvgPool2d

Strategy: Direct Mapping

Official Docs ↗