Searched refs:NestedModulePDTWrapper (Results 1 – 1 of 1) sorted by relevance
68 class NestedModulePDTWrapper(torch.nn.Module): class76 make_global(NestedPDTInner, NestedModulePDTWrapper)78 wrapped_pdt_model = NestedModulePDTWrapper(inner_pdt_model)