Searched refs:CheckpointTwiceModuleWeightSharing (Results 1 – 1 of 1) sorted by relevance
497 class CheckpointTwiceModuleWeightSharing(CheckpointTwiceModule): class in CommonDistributedDataParallelTest723 self.CheckpointTwiceModuleWeightSharing(),