| 24 | |
| 25 | |
| 26 | class SimpleConv(Conv): |
| 27 | |
| 28 | def __call__(self, x, edge_index, size=None): |
| 29 | gather_x, = self.gather_feature([x], edge_index) |
| 30 | out = self.apply_edge(gather_x[1]) |
| 31 | out = mp_ops.scatter_(self.aggr, out, edge_index[0], size=size[0]) |
| 32 | out = self.apply_node(out) |
| 33 | return out |
| 34 | |
| 35 | |
| 36 | class ConvTest(tf.test.TestCase): |