Searched refs:assertGraphContainsExactly (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit_llga_fuser.py | 93 self.assertGraphContainsExactly(graph, pat, 0)
161 self.assertGraphContainsExactly(graph, LLGA_FUSION_GROUP, 1)
170 self.assertGraphContainsExactly(graph, LLGA_FUSION_GROUP, 0)
189 self.assertGraphContainsExactly(graph, LLGA_FUSION_GROUP, 0)
210 self.assertGraphContainsExactly(graph, LLGA_FUSION_GROUP, 1)
231 self.assertGraphContainsExactly(graph, LLGA_FUSION_GROUP, 1)
246 self.assertGraphContainsExactly(graph, LLGA_FUSION_GROUP, 0)
256 self.assertGraphContainsExactly(graph, LLGA_FUSION_GROUP, 0)
265 self.assertGraphContainsExactly(graph, LLGA_FUSION_GROUP, 1)
289 self.assertGraphContainsExactly(graph, LLGA_FUSION_GROUP, 1)
[all …]
|
| H A D | test_jit_fuser.py | 285 self.assertGraphContainsExactly(forward_graph, 'prim::FusionGroup', 1)
704 self.assertGraphContainsExactly(
728 self.assertGraphContainsExactly(
772 self.assertGraphContainsExactly(forward_graph, 'prim::FusionGroup', 1)
811 self.assertGraphContainsExactly(
933 self.assertGraphContainsExactly(
|
| H A D | test_mkldnn_fusion.py | 30 self.assertGraphContainsExactly(graph, pat, 0)
98 self.assertGraphContainsExactly(graph, FUSION_GROUP, 1)
127 self.assertGraphContainsExactly(graph, FUSION_GROUP, 1)
|
| H A D | test_jit_fuser_te.py | 972 self.assertGraphContainsExactly(ge.graph_for(*inputs), FUSION_GROUP, 3, True)
1037 self.assertGraphContainsExactly(
1068 self.assertGraphContainsExactly(
1275 self.assertGraphContainsExactly(
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_autodiff_subgraph_slicing.py | 106 self.assertGraphContainsExactly(
109 self.assertGraphContainsExactly(
318 self.assertGraphContainsExactly(graph, "prim::DifferentiableGraph", 1)
333 self.assertGraphContainsExactly(graph, "prim::DifferentiableGraph", 1)
345 self.assertGraphContainsExactly(graph, "prim::DifferentiableGraph", 2)
360 self.assertGraphContainsExactly(graph, "prim::DifferentiableGraph", 1)
377 self.assertGraphContainsExactly(graph, "prim::DifferentiableGraph", 1)
390 self.assertGraphContainsExactly(graph, "prim::DifferentiableGraph", 2)
405 self.assertGraphContainsExactly(graph, "prim::DifferentiableGraph", 1)
423 self.assertGraphContainsExactly(graph, "prim::DifferentiableGraph", 1)
[all …]
|
| H A D | test_async.py | 293 self.assertGraphContainsExactly(
297 self.assertGraphContainsExactly(
300 self.assertGraphContainsExactly(
400 self.assertGraphContainsExactly(
403 self.assertGraphContainsExactly(
406 self.assertGraphContainsExactly(
442 self.assertGraphContainsExactly(
445 self.assertGraphContainsExactly(
448 self.assertGraphContainsExactly(
|
| H A D | test_await.py | 176 self.assertGraphContainsExactly(
214 self.assertGraphContainsExactly(
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | jit_utils.py | 315 def assertGraphContainsExactly(self, graph, kind, num_kind_nodes, consider_subgraphs=False): member in JitTestCase
|