Searched refs:NestedModulePDTOuter (Results 1 – 1 of 1) sorted by relevance
94 class NestedModulePDTOuter(torch.nn.Module): class102 make_global(NestedModulePDTInner, NestedModulePDTOuter)104 outer_pdt_model = NestedModulePDTOuter(inner_pdt_model)