Conv2d

2D Convolution

Abstract Signature:

Conv2d(in_channels, out_channels, kernel_size)

PyTorch

API: torch.nn.Conv2d
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.layers.Conv2D
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: keras.layers.Conv2D
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.nn.layers.convolution.Conv2d
Strategy: Direct Mapping
Official Docs ↗

LaTeX DSL (MIDL)

API: midl.Conv2d
Strategy: Direct Mapping

NVIDIA SASS

API: Macro.Conv2d
Strategy: Direct Mapping

HTML Grid DSL

API: html_dsl.Conv2d
Strategy: Direct Mapping

Flax NNX

API: flax.nnx.Conv
Strategy: Direct Mapping
Official Docs ↗