Home
last modified time | relevance | path

Searched refs:flatten_graph_inputs (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/backends/
H A Dcommon.py10 from torch._dynamo.utils import counters, defake, flatten_graph_inputs
28 return flatten_graph_inputs(
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcompile_fx.py30 flatten_graph_inputs,
1335 return flatten_graph_inputs(
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dutils.py2710 def flatten_graph_inputs(gm: torch.fx.GraphModule, inputs, compile_gm): function