Searched refs:flatten_graph_inputs (Results 1 – 3 of 3) sorted by relevance
10 from torch._dynamo.utils import counters, defake, flatten_graph_inputs28 return flatten_graph_inputs(
30 flatten_graph_inputs,1335 return flatten_graph_inputs(
2710 def flatten_graph_inputs(gm: torch.fx.GraphModule, inputs, compile_gm): function