code: def _conv_fn(
      name: str,
      module: nn.Module,
      device_mesh: DeviceMesh,
  ) -> None:
      for name, param in module.named_parameters():
          dist_spec = [Replicate()]
          dist_param = torch.nn.Parameter(
              distribute_tensor(param, device_mesh, dist_spec)
          )
          name = "_".join(name.split("."))
          module.register_parameter(name, dist_param)
  
file_path: pytorch/test/distributed/_tensor/test_convolution_ops.py
imports: import copy
  import torch
  import torch.nn
  from torch.distributed._tensor import DeviceMesh, distribute_module, distribute_tensor, Replicate, Shard
  from torch.testing._internal.common_utils import run_tests
  from torch.testing._internal.distributed._tensor.common_dtensor import DTensorTestBase, with_comms
label_desc: Use image node_id 1 for calling a global function with example usage: _conv_fn(name, module, device_mesh) without return types