def _prepare_init(self, process_group=None, init_rrefs=False):
    self._init_rrefs = init_rrefs
    self._sharded_tensor_id = None
    self._process_group = (process_group if (process_group is not None) else distributed_c10d._get_default_group())
    self._local_shards: List[Shard] = []
    self._remote_shards: Dict[(int, List[rpc.RRef[Shard]])] = {}
def _prepare_init(self, process_group=None, init_rrefs=False):
    self._init_rrefs = init_rrefs
    self._sharded_tensor_id = None
    self._process_group = (process_group if (process_group is not None) else distributed_c10d._get_default_group())
    self._local_shards: List[Shard] = []
    self._remote_shards: Dict[(int, List[rpc.RRef[Shard]])] = {}