Method forward

tests/python/relax/test_frontend_from_fx.py:4319–4320

(self, input)

    class Cumsum(Module):
        def forward(self, input):
            return torch.cumsum(input, dim=1, dtype=torch.int32)

    @tvm.script.ir_module
    class expected1: