Searched refs:assertGraphContains (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/pytorch/test/ |
H A D | test_mkldnn_fusion.py | 100 self.assertGraphContains(graph, kind=conv_node_name) 129 self.assertGraphContains(graph, kind='aten::conv2d') 170 self.assertGraphContains(graph, kind='aten::_convolution')
|
H A D | test_jit_llga_fuser.py | 111 self.assertGraphContains(fusion_groups[i], pattern) 750 self.assertGraphContains(t_jit.graph_for(x, y), LLGA_FUSION_GROUP) 759 self.assertGraphContains(t_jit_2.graph_for(x, y), LLGA_FUSION_GROUP)
|
H A D | test_jit.py | 1986 self.assertGraphContains(jitted.graph, kind='aten::einsum')
|
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/utils/ |
H A D | testing.py | 129 def assertGraphContains(self, op_regex, n): member in AutoGraphTestCase
|
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/ |
H A D | control_flow_test.py | 667 self.assertGraphContains(r'(while/)?pow$', 1) 742 self.assertGraphContains(r'(while/)?pow$', 1)
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | jit_utils.py | 295 def assertGraphContains(self, graph, kind, consider_subgraphs=False): member in JitTestCase
|