Searched refs:_check_map_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 | 1183 graphs = self._check_map_graph_and_extract(map_f, (xs, y)) 1215 graphs = self._check_map_graph_and_extract(f, (x,)) 1251 graphs = self._check_map_graph_and_extract(f, (x,)) 1300 graphs = self._check_map_graph_and_extract(fn, (x, y)) 1334 graphs = self._check_map_graph_and_extract(fn, (x, y)) 1558 def _check_map_graph_and_extract(self, fn, args): member in HigherOrderOpTests
|