|
|
def | __init__ (self, module, device_ids=None, output_device=None, dim=0, broadcast_buffers=True, process_group=None, bucket_cap_mb=25, check_reduction=False) |
| |
|
def | __getstate__ (self) |
| |
|
def | __setstate__ (self, state) |
| |
|
def | forward (self, inputs, kwargs) |
| |
|
def | scatter (self, inputs, kwargs, device_ids) |
| |
|
def | parallel_apply (self, replicas, inputs, kwargs) |
| |
|
def | gather (self, outputs, output_device) |
| |
|
def | train (self, mode=True) |
| |
|
|
| process_group |
| |
|
| dim |
| |
|
| module |
| |
|
| device_ids |
| |
|
| output_device |
| |
|
| broadcast_buffers |
| |
|
| check_reduction |
| |
|
| broadcast_bucket_size |
| |
|
| bucket_bytes_cap |
| |
|
| modules_params_data |
| |
|
| modules_buffers_data |
| |
|
| bucket_sizes |
| |
|
| bucket_map |
| |
|
| buckets |
| |
|
| buckets_ready_size |
| |
|
| buckets_coalesced |
| |
|
| next_bucket |
| |
|
| all_buckets_reduced |
| |
|
| check_previous_reduction |
| |
|
| ready_buckets_not_reduced |
| |
|
| reduction_works |
| |
|
| devs_ready |
| |
|
| default_streams |
| |
Definition at line 19 of file distributed.py.
The documentation for this class was generated from the following file:
1.8.11
