    def __init__(self, stride, pad, dilate=1, group=1) -> None:
        """Conv2d constructor.

        Parameters
        ----------
        stride : int or tuple of two ints
            Stride of the convolution. If tuple is
            provided, the elements correspond to height and width stride
            respectively
        pad : int or tuple of two ints
            Size of padding to add to the input. If tuple is
            provided, the elements correspond to height and width padding
            respectively
        dilate : int or tuple of two ints, optional
            Size of spacing between kernel elements, by default 1. If tuple is
            provided, the elements correspond to height and width dilation
            respectively
        group : int, optional
           Number of blocked connections from input
            channels to output channels, by default 1
        """
        super().__init__()
        self._attrs["op"] = "conv2d"
        self._attrs["stride"] = stride
        self._attrs["pad"] = pad
        self._attrs["dilate"] = dilate
        self._attrs["group"] = group
        self._attrs["has_profiler"] = True
        self._attrs["epilogue_alignment"] = 1
        self._attrs["epilogue"] = "LinearCombination"
        self._attrs["workspace"] = 0
        self._attrs["split_k"] = None
        self.shape_eval_template = SHAPE_FUNC_TEMPLATE
        self.shape_save_template = SHAPE_ASSIGNMENT_TEMPLATE
        self.exec_key_template = EXEC_KEY_TEMPLATE
        self.exec_dyn_key_template = EXEC_DYN_KEY_TEMPLATE
        self.exec_cond_template = EXEC_COND_TEMPLATE
Conv2d constructor.

Parameters
----------
stride : int or tuple of two ints
Stride of the convolution. If tuple is
provided, the elements correspond to height and width stride
respectively
pad : int or tuple of two ints
Size of padding to add to the input. If tuple is
provided, the elements correspond to height and width padding
respectively
dilate : int or tuple of two ints, optional
Size of spacing between kernel elements, by default 1. If tuple is
provided, the elements correspond to height and width dilation
respectively
group : int, optional
Number of blocked connections from input
channels to output channels, by default 1
python/aitemplate/compiler/ops/conv/conv2d.py