(self, x, edge_index, size=None, **kwargs)
| 41 | return tuple(map(deg_inv_sqrt, [0, 1])) |
| 42 | |
| 43 | def __call__(self, x, edge_index, size=None, **kwargs): |
| 44 | norm = self.norm(edge_index, size) |
| 45 | hidden = x |
| 46 | for k in range(self.K): |
| 47 | x, gather_norm, = self.gather_feature([x, norm], edge_index) |
| 48 | out = self.apply_edge(x[1], gather_norm[0], gather_norm[1]) |
| 49 | out = mp_ops.scatter_(self.aggr, out, edge_index[0], size=size[0]) |
| 50 | out = self.apply_node(out, hidden[0]) |
| 51 | x = [out, hidden[1]] |
| 52 | return out |
| 53 | |
| 54 | def apply_edge(self, x_j, norm_i, norm_j): |
| 55 | return norm_i * norm_j * x_j |
nothing calls this directly
no test coverage detected