def METHOD_NAME(test_case, device):
    x = flow.tensor(3.0, device=device)
    class MyModule(flow.nn.Module):
        def __init__(self):
            super().__init__()
            self.weight = flow.nn.Parameter(flow.tensor(5.0, device=device))
        def forward(self, x):
            return x * self.weight + 1.0
    my_module = MyModule()
    of_eager_out = my_module(x)
    class ScalarGraph(flow.nn.Graph):
        def __init__(self):
            super().__init__()
            self.m = my_module
        def build(self, x):
            return self.m(x)
    scalar_g = ScalarGraph()
    of_lazy_out = scalar_g(x)
    test_case.assertTrue(np.array_equal(of_lazy_out.numpy(), of_eager_out.numpy()))