Searched refs:DynamicCheckpointTwiceModule (Results 1 – 1 of 1) sorted by relevance
513 class DynamicCheckpointTwiceModule(CheckpointTwiceModule): class in CommonDistributedDataParallelTest527 class DynamicCheckpointTwiceModuleWeightSharing(DynamicCheckpointTwiceModule):660 self.DynamicCheckpointTwiceModule(use_reentrant=False),