| 17 | |
| 18 | |
| 19 | class Simple(Module): |
| 20 | def __init__(self): |
| 21 | super().__init__() |
| 22 | self.a = Parameter([1.23], dtype="float32") |
| 23 | |
| 24 | def forward(self, x): |
| 25 | x = x * self.a |
| 26 | return x |
| 27 | |
| 28 | |
| 29 | @pytest.mark.parametrize("format", ["chrome_timeline.json", "memory_flow.svg"]) |