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)`` .. raw:: html

PyTorch

API: torch.mkldnn_convolution
Strategy: Direct Mapping

JAX (Core)

API: jax.lax.conv_general_dilated
Strategy: Plugin (conv2d_functional)

Keras

API: keras.ops.conv
Strategy: Direct Mapping

Apple MLX

API: mlx.nn.Conv2d
Strategy: Direct Mapping