Searched refs:sdpa_constraint (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 2162 def sdpa_constraint(fx_node, *args, **kwargs): function
2373 sdpa_constraint,
2378 sdpa_constraint,
2383 sdpa_constraint,
2388 sdpa_constraint,
2393 sdpa_constraint,
2398 sdpa_constraint,
2403 sdpa_constraint,
2408 sdpa_constraint,
2411 make_fallback(aten._flash_attention_forward.default, sdpa_constraint)
[all …]
|