Searched refs:_check_cond_graph_and_extract (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/test/dynamo/ |
H A D | test_higher_order_ops.py | 1540 def _check_cond_graph_and_extract(self, fn, args): member in HigherOrderOpTests 1587 graphs = self._check_cond_graph_and_extract(fn, (torch.randn(4, 5),)) 1630 self._check_cond_graph_and_extract(fn, (torch.randn(4, 5),)) 1631 graphs = self._check_cond_graph_and_extract(fn, (torch.randn(4, 5),))
|