Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_mkldnn_fusion.py100 self.assertGraphContains(graph, kind=conv_node_name)
129 self.assertGraphContains(graph, kind='aten::conv2d')
170 self.assertGraphContains(graph, kind='aten::_convolution')
H A Dtest_jit_llga_fuser.py111 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 Dtest_jit.py1986 self.assertGraphContains(jitted.graph, kind='aten::einsum')
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/utils/
H A Dtesting.py129 def assertGraphContains(self, op_regex, n): member in AutoGraphTestCase
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/
H A Dcontrol_flow_test.py667 self.assertGraphContains(r'(while/)?pow$', 1)
742 self.assertGraphContains(r'(while/)?pow$', 1)
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Djit_utils.py295 def assertGraphContains(self, graph, kind, consider_subgraphs=False): member in JitTestCase