Searched refs:graph_call_function (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | reinplace.py | 45 def graph_call_function(graph: torch.fx.Graph, fn, *args, **kwargs): function
100 view = graph_call_function(
105 return graph_call_function(
153 inp = graph_call_function(graph, aten.clone, inp)
157 …tmp = graph_call_function(graph, view.target, tmp, *view.args, **view.kwargs) # type: ignore[unio…
159 graph_call_function(graph, aten.copy_.default, tmp, src)
284 new_node = graph_call_function(
297 new_node = graph_call_function(
308 new_src = graph_call_function(
|