Searched refs:ForwardReturnValueModule (Results 1 – 1 of 1) sorted by relevance
1056 class ForwardReturnValueModule(nn.Module): class1081 ForwardReturnValueModule().float().to(device_id),