    def forward(self, *args):
        """forward pass for calling T5 block"""
        assert len(args) == 1
        x = args[0]
        hidden = self.op(x, self.wi_0_weight.tensor(), self.wi_1_weight.tensor())
        return self.wo(hidden)
forward pass for calling T5 block
python/aitemplate/frontend/nn/dual_gemm.py