Method calc_conv

imperative/python/megengine/module/conv.py:414–442

(self, inp, weight, bias)

        )

    def calc_conv(self, inp, weight, bias):
        assert self.padding_mode in [
            "zeros",
            "reflect",
            "replicate",
        ]
        if self.padding_mode != "zeros":
            return conv2d(
                pad(inp, self.get_pad_witdth(), self.padding_mode),
                weight,
                bias,
                self.stride,
                0,
                self.dilation,
                self.groups,
                self.conv_mode,
                self.compute_mode,
            )
        return conv2d(
            inp,
            weight,
            bias,
            self.stride,
            self.padding,
            self.dilation,
            self.groups,
            self.conv_mode,
            self.compute_mode,
        )

    def forward(self, inp):
        return self.calc_conv(inp, self.weight, self.bias)