PyTorch Forums
Script function as target
torch.compile
FX (Functional Transformations)
Yuchen_Wang
(Yuchen Wang)
January 17, 2023, 2:49am
1
Is it possible to have a torch.jit script function as node target in torch.fx?