Searched defs:combine_mode (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_control_flow.py | 1223 def test_pointwise_associative_scan_simple(self, reverse, combine_mode, device): argument
1269 def test_pointwise_associative_scan_dim(self, reverse, combine_mode, device): argument
1312 self, reverse, combine_mode, compile_mode, device argument
1373 self, reverse, combine_mode, device argument
1425 def test_pointwise_associative_scan_tuple(self, reverse, combine_mode, device): argument
1455 self, reverse, combine_mode, device argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_control_flow.py | 711 def test_associative_scan_CUDA_flip(self, combine_mode, backend, device): argument
|