Searched refs:DataclassOutputModule (Results 1 – 1 of 1) sorted by relevance
977 class DataclassOutputModule(nn.Module): class in CommonDistributedDataParallelTest998 net = self.DataclassOutputModule(skip_o1=skip_o1).to(self.rank)