Searched refs:runGraph (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/test/cpp/jit/ |
| D | test_concat_opt.cpp | 33 auto orig_outputs = runGraph(graph, inputs); in TEST()
37 auto opt_outputs = runGraph(graph, inputs); in TEST()
78 auto orig_outputs = runGraph(graph, inputs); in TEST()
82 auto opt_outputs = runGraph(graph, inputs); in TEST()
129 auto orig_outputs = runGraph(graph, inputs); in TEST()
133 auto opt_outputs = runGraph(graph, inputs); in TEST()
167 auto orig_outputs = runGraph(graph, inputs); in TEST()
171 auto opt_outputs = runGraph(graph, inputs); in TEST()
201 auto orig_outputs = runGraph(graph, inputs); in TEST()
205 auto opt_outputs = runGraph(graph, inputs); in TEST()
[all …]
|
| D | test_stack_opt.cpp | 37 auto orig_outputs = runGraph(graph, inputs); in TEST()
41 auto opt_outputs = runGraph(graph, inputs); in TEST()
82 auto orig_outputs = runGraph(graph, inputs); in TEST()
86 auto opt_outputs = runGraph(graph, inputs); in TEST()
122 auto orig_outputs = runGraph(graph, inputs); in TEST()
126 auto opt_outputs = runGraph(graph, inputs); in TEST()
165 auto orig_outputs = runGraph(graph, inputs); in TEST()
169 auto opt_outputs = runGraph(graph, inputs); in TEST()
210 auto orig_outputs = runGraph(graph, inputs); in TEST()
215 auto opt_outputs = runGraph(graph, inputs); in TEST()
[all …]
|
| D | test_utils.h | 92 std::vector<at::Tensor> runGraph(
|
| D | test_utils.cpp | 281 std::vector<at::Tensor> runGraph( in runGraph() function
|