Method get_graph_pool

tf_euler/python/mp_utils/graph_gnn.py:56–57

(self, graph_pool)

        raise NotImplementedError

    def get_graph_pool(self, graph_pool):
        return graph_pool('add')

    def get_conv(self, conv_class, dim):
        return conv_class(dim)