Searched defs:inv_scale_factor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | fuse_attention.py | 77 def _sfdp_pattern_3(query, key, value, inv_scale_factor, dropout_p): argument
86 def _sfdp_replacement_3(query, key, value, inv_scale_factor, dropout_p): argument
298 def _sfdp_pattern_12(query, key, value, inv_scale_factor, dropout_p): argument
308 def _sfdp_replacement_12(query, key, value, inv_scale_factor, dropout_p): argument
|