(self, xs, y, z)
| 230 | |
| 231 | class MapModule(torch.nn.Module): |
| 232 | def forward(self, xs, y, z): |
| 233 | def body(x, y, z): |
| 234 | return x + y + z |
| 235 | |
| 236 | return torch_map(body, xs, y, z) |
| 237 | |
| 238 | inps = (torch.ones(6, 4), torch.tensor(5), torch.tensor(4)) |
| 239 | exported_map_example = export(MapModule(), inps) |
nothing calls this directly
no outgoing calls
no test coverage detected