functorch.compile.ts_compile# functorch.compile.ts_compile(fx_g, inps)[source]# Compiles the fx_g with Torchscript compiler. Warning This API is experimental and likely to change. Parameters fx_g (fx.GraphModule) – The input Fx graph module to be compiled. Returns Torch scripted model.
functorch.compile.ts_compile# functorch.compile.ts_compile(fx_g, inps)[source]# Compiles the fx_g with Torchscript compiler. Warning This API is experimental and likely to change. Parameters fx_g (fx.GraphModule) – The input Fx graph module to be compiled. Returns Torch scripted model.