(self, x, edge_index, size=None, edge_attr=None, **kwarg)
| 51 | self.built = True |
| 52 | |
| 53 | def __call__(self, x, edge_index, size=None, edge_attr=None, **kwarg): |
| 54 | assert edge_attr is not None |
| 55 | if not self.built: |
| 56 | self.build() |
| 57 | gather_x, = self.gather_feature([x], edge_index) |
| 58 | out = self.apply_edge(gather_x[1], edge_attr) |
| 59 | out = mp_ops.scatter_(self.aggr, out, edge_index[0], size=size[0]) |
| 60 | out = self.apply_node(x[0], out) |
| 61 | return out |
| 62 | |
| 63 | def apply_edge(self, x_j, edge_attr): |
| 64 | edge_attr, idx = tf.unique(edge_attr) |
nothing calls this directly
no test coverage detected