Enable dataflow prime and add edge_aggregation_method choice to example.
Enable dataflow prime and add edge_aggregation_method choice to example.

PiperOrigin-RevId: 462134687

tensorflow/gnn,tensorflow/gnn