Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.proxy_tensor.selective_decompose#

torch.fx.experimental.proxy_tensor.selective_decompose(joint_gm, *args, decomposition, should_decompose, trace_joint_graph)[source]#

Retrace a joint graph module and selectively apply decomposition.

Return type:

GraphModule

On this page
Show Source
PyTorch Libraries