code: def forward(self, *inputs, **kwargs):
      self.needs_reduction = True
      return self.module(*inputs, **kwargs)
  
file_path: DALI/docs/examples/use_cases/video_superres/common/distributed.py
imports: import torch
  from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors
  import torch.distributed
  from torch.nn.modules import Module
label_desc: Use image node_id 2 for calling the DistributedDataParallel obj's underlying member method code with example usage: obj.forward() and returns: self