Searched refs:TestPartitionFunctions (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fx_passes.py | 63 class TestPartitionFunctions: class
255 …(TestPartitionFunctions.forward1, [["add_7", "add_6"], ["add_5", "add_4", "add_3"], ["add_2", "add…
256 (TestPartitionFunctions.forward2, [["add_3", "add_2"], ["add_1", "add"]], False),
259 (TestPartitionFunctions.forward3, [["add_2", "add_1", "add"]], False),
260 (TestPartitionFunctions.forward4, [["add_2", "add_1", "add"]], False),
263 (TestPartitionFunctions.forward5, [["add_1", "add"]], False),
264 (TestPartitionFunctions.forward6, [["add"]], False),
265 (TestPartitionFunctions.forward7, [["add_3", "add_2", "add", "add_1"]], False),
266 (TestPartitionFunctions.forward8, [["add_3", "add_2", "add", "add_1"]], False),
269 (TestPartitionFunctions.forward9, [['add_3', 'add_2', 'add_1', 'add']], False),
[all …]
|