Function _normalize_conv_param

backends/mlx/ops.py:2561–2569

Normalize a conv parameter (stride/padding/etc.) to a list of length ndim.

(val, ndim, default=0)

def _normalize_conv_param(val, ndim, default=0):
    """Normalize a conv parameter (stride/padding/etc.) to a list of length ndim."""
    if isinstance(val, int):
        return [val] * ndim
    if isinstance(val, list):
        if len(val) == 1:
            return val * ndim
        return val
    return [default] * ndim


@REGISTRY.register(target=[torch.ops.aten.convolution.default])