flex_attention (definition) in projects: aosp_15_r20

Searched defs:flex_attention (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/kernel/
H A D	flex_attention.py	`674 def flex_attention( function`
/aosp_15_r20/external/pytorch/torch/nn/attention/
H A D	flex_attention.py	`916 def flex_attention( function`
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A D	flex_attention.py	`122 flex_attention = FlexAttentionHOP() variable`