Method forward

profiler/test/test_profiler_e2e.py:40–43

(self, x)

        self.register_buffer("b", 2 * torch.ones(2, 2, dtype=torch.float))

    def forward(self, x):
        a = torch.mul(self.a, x)
        b = torch.add(a, self.b)
        return b


class TestCustomOps(unittest.TestCase):