    def forward(self, *args):
        """Applies Conv3d on the input tensor."""
        assert len(args) == 1
        x = args[0]

        if self.has_bias:
            x = ndhwc3to8()(x)
            weight = ndhwc3to8()(self.weight.tensor())
            return self.op(x, weight, self.bias.tensor())
        else:
            return self.op(x, self.weight.tensor())
Applies Conv3d on the input tensor.
python/aitemplate/frontend/nn/conv3d.py