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
jax.lax.conv_general_dilated