Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_input_attr_tracking.py28 def grab_graph_backend(gm, inps): function
33 fn = torch._dynamo.optimize(grab_graph_backend, nopython=True)(fn)
57 def grab_graph_backend(gm, inps): function
62 fn = torch._dynamo.optimize(grab_graph_backend, nopython=True)(fn)
86 def grab_graph_backend(gm, inps): function
91 fn = torch._dynamo.optimize(grab_graph_backend, nopython=True)(fn)