Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_profiler.py62 g = torch.jit.last_executed_optimized_graph()
77 g = torch.jit.last_executed_optimized_graph()
97 g = torch.jit.last_executed_optimized_graph()
117 g = torch.jit.last_executed_optimized_graph()
133 g = torch.jit.last_executed_optimized_graph()
155 g = torch.jit.last_executed_optimized_graph()
174 g = torch.jit.last_executed_optimized_graph()
185 g = torch.jit.last_executed_optimized_graph()
196 g = torch.jit.last_executed_optimized_graph()
213 g = torch.jit.last_executed_optimized_graph()
[all …]
H A Dtest_complexity.py80 g = torch.jit.last_executed_optimized_graph()
100 g = torch.jit.last_executed_optimized_graph()
/aosp_15_r20/external/pytorch/torch/jit/
H A D_fuser.py84 last_executed_optimized_graph = torch._C._last_executed_optimized_graph variable
128 return last_executed_optimized_graph()
H A D__init__.py29 last_executed_optimized_graph,
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit_fuser_te.py201 self.assertAllFused(torch.jit.last_executed_optimized_graph())
1214 g = torch.jit.last_executed_optimized_graph()
1330 torch.jit.last_executed_optimized_graph()
1341 torch.jit.last_executed_optimized_graph()
1352 torch.jit.last_executed_optimized_graph()
1365 torch.jit.last_executed_optimized_graph()
2181 torch.jit.last_executed_optimized_graph()
2246 g = torch.jit.last_executed_optimized_graph()
2262 g = torch.jit.last_executed_optimized_graph()
2458 g = torch.jit.last_executed_optimized_graph()
[all …]
H A Dtest_jit.py3160 g = torch.jit.last_executed_optimized_graph()
3178 g = torch.jit.last_executed_optimized_graph()
3205 g = torch.jit.last_executed_optimized_graph()
3224 graph_str = torch.jit.last_executed_optimized_graph()
6272 g = torch.jit.last_executed_optimized_graph()
6280 g = torch.jit.last_executed_optimized_graph()
6298 g = torch.jit.last_executed_optimized_graph()
6316 g = torch.jit.last_executed_optimized_graph()
6324 g = torch.jit.last_executed_optimized_graph()
6342 g = torch.jit.last_executed_optimized_graph()
[all …]
H A Dtest_jit_autocast.py766 g = torch.jit.last_executed_optimized_graph()
H A Dtest_tensorexpr.py28 self.assertAllFused(torch.jit.last_executed_optimized_graph())
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dgraph_executor.cpp105 thread_local std::weak_ptr<Graph> last_executed_optimized_graph; variable
107 return last_executed_optimized_graph.lock(); in lastExecutedOptimizedGraph()
587 last_executed_optimized_graph = plan.graph; in run()
613 last_executed_optimized_graph = frame->plan.graph; in runAsync()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/
H A Dtutorial.md52 print(torch.jit.last_executed_optimized_graph())