Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_higher_order_ops.py2257 def _get_source_fn_stack(self, gm, node_names): member in HigherOrderOpTests
2290 actual_stack = self._get_source_fn_stack(gm, {"cos", "add", "linear"})
2325 actual_stack = self._get_source_fn_stack(gm, {"cos", "add", "sin", "sub"})
2354 actual_stack = self._get_source_fn_stack(gm, {"cos", "add", "sin"})
2374 actual_stack = self._get_source_fn_stack(gm, {"sum_1", "sin"})
2410 actual_stack = self._get_source_fn_stack(