Shortcuts

functorch.compile.ts_compile

functorch.compile.ts_compile(fx_g, inps)[source]

使用Torchscript编译器编译fx_g

警告

此API是实验性的,可能会发生变化。

Parameters

fx_g (fx.GraphModule) – 要编译的输入Fx图模块。

Returns

Torch脚本化模型。