dgl.contrib.graph_store.SharedMemoryDGLGraph.update_all¶
-
SharedMemoryDGLGraph.
update_all
(message_func='default', reduce_func='default', apply_node_func='default')[source]¶ Distribute the computation in update_all among all pre-defined workers.
update_all requires that all workers invoke this method and will return only when all workers finish their own portion of computation. The number of workers are pre-defined. If one of them doesn’t invoke the method, it won’t return because some portion of computation isn’t finished.
- Parameters
message_func (callable, optional) – Message function on the edges. The function should be an
Edge UDF
.reduce_func (callable, optional) – Reduce function on the node. The function should be a
Node UDF
.apply_node_func (callable, optional) – Apply function on the nodes. The function should be a
Node UDF
.