Home
last modified time | relevance | path

Searched defs:num_fwd_outputs (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_functorch/
H A Dpartitioners.py268 joint_module: fx.GraphModule, *, num_fwd_outputs argument
385 joint_module: fx.GraphModule, _joint_inputs, *, num_fwd_outputs argument
1733 num_fwd_outputs, argument