Function _BroadcastComputedParams

caffe2/python/data_parallel_model.py:1515–1519

(devices, model, rendezvous, use_nccl=False)

def _BroadcastComputedParams(devices, model, rendezvous, use_nccl=False):
    if rendezvous is None:
        _BroadcastComputedParamsSingleHost(devices, model, use_nccl)
    else:
        _BroadcastComputedParamsDistributed(devices, model, rendezvous, use_nccl)


def _BroadcastComputedParamsDistributed(