Searched refs:CPUGPUModule (Results 1 – 1 of 1) sorted by relevance
769 class CPUGPUModule(nn.Module): class775 cpu_gpu = CPUGPUModule()782 FSDP(CPUGPUModule())788 class CPUGPUModule(nn.Module): class795 m = CPUGPUModule()802 m = CPUGPUModule()816 class CPUGPUModule(nn.Module): class823 m = CPUGPUModule()