Searched defs:GmWrapper (Results 1 – 2 of 2) sorted by relevance
452 mod: Union[torch.fx.GraphModule, torch._dynamo.utils.GmWrapper],
2699 class GmWrapper(torch.nn.Module): class