Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_sdpa.py11 def allow_in_graph_sdpa_params(): function
29 with allow_in_graph_sdpa_params():
46 with allow_in_graph_sdpa_params():
65 with allow_in_graph_sdpa_params():
83 with allow_in_graph_sdpa_params():