(self)
| 42 | |
| 43 | class CustomModule(flow.nn.Module): |
| 44 | def __init__(self): |
| 45 | super().__init__() |
| 46 | self.layer = SubModule() |
| 47 | self.fc1 = flow.nn.Linear(36, 4) |
| 48 | self.register_buffer("dummy_buff", flow.Tensor(1, 4)) |
| 49 | |
| 50 | def forward(self, x): |
| 51 | x = self.layer(x) |
nothing calls this directly
no test coverage detected