MCPcopy Index your code
hub / github.com/pytorch/pytorch / broadcast

Function broadcast

caffe2/python/data_parallel_model.py:1199–1207  ·  view source on GitHub ↗
(params)

Source from the content-addressed store, hash-verified

1197 params_group = list(model._device_grouped_blobs[param_name].values())
1198
1199 def broadcast(params):
1200 comm_world, control_input = context.get_control_and_context(params)
1201 net.Broadcast(
1202 inputs=[comm_world] + params,
1203 outputs=params,
1204 name=param_name,
1205 engine=rendezvous['engine'],
1206 control_input=control_input
1207 )
1208
1209 device_opt = gpu_device_opt if _IsGPUBlob(
1210 model, param_name

Callers 1

Calls 1

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…