SDPA (definition) in projects: external

Searched defs:SDPA (Results 1 – 4 of 4) sorted by relevance

/external/executorch/backends/xnnpack/partition/graphs/
D	sdpa.py	`23 class SDPA(torch.nn.Module): class`
/external/executorch/backends/xnnpack/test/ops/
D	sdpa.py	`18 class SDPA(torch.nn.Module): class in TestSDPA`
/external/executorch/extension/llm/modules/
D	attention.py	`317 class SDPA(nn.Module): class`
/external/executorch/examples/models/llama/
D	llama_transformer.py	`214 class SDPA(nn.Module): class`