Searched refs:unbind_stack_to_slices (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_split_cat_fx_passes.py | 1359 def unbind_stack_to_slices(x): function
1454 (unbind_stack_to_slices, 0, 0, 0, 0, 0, 1, 0),
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | split_cat.py | 2308 def unbind_stack_to_slices(match: Match, unbind_input: torch.fx.Node, dim: int): function
|