    def forward(self, *args):
        """Applies Conv2d on the input tensor."""
        assert len(args) == 1
        x = args[0]
        return self.op(x, self.weight.tensor())
Applies Conv2d on the input tensor.
python/aitemplate/frontend/nn/conv2d/conv2d.py