MkldnnConvolution

Applies 2D convolution (MKLDNN backend optimization).

Abstract Signature:

MkldnnConvolution(input: Tensor, weight: Tensor, bias: Tensor, padding: int = 0, stride: int = 1, dilation: int = 1, groups: int = 1)

PyTorchJAX (Core)KerasApple MLX

PyTorch

API: torch.mkldnn_convolution

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.lax.conv_general_dilated

Strategy: Plugin (conv2d_functional)

Keras

API: keras.ops.conv

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.Conv2d

Strategy: Direct Mapping

Official Docs ↗