(self, *args)
| 64 | return x |
| 65 | |
| 66 | def forward(self, *args): |
| 67 | if isinstance(args[0], Graph): |
| 68 | return self.mini_forward(*args) |
| 69 | else: |
| 70 | device = next(self.parameters()).device |
| 71 | x, adjs = args |
| 72 | for i, (src_id, graph, size) in enumerate(adjs): |
| 73 | graph = graph.to(device) |
| 74 | output = self.convs[i](graph, x) |
| 75 | x = output[: size[1]] |
| 76 | if i != self.num_layers - 1: |
| 77 | x = F.relu(x) |
| 78 | x = F.dropout(x, p=self.dropout, training=self.training) |
| 79 | return x |
| 80 | |
| 81 | def inference(self, x_all, data_loader): |
| 82 | device = next(self.parameters()).device |
nothing calls this directly
no test coverage detected