For graph propagation, we use Graph Attention Network (GAT)  (Velickovic et al., 2018) to perform message passing over the hierarchical graph.
gAT takes all the nodes as input, and updates node feature h i through its neighbors N i in the graph.