Home
last modified time | relevance | path

Searched defs:Attention (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/
Ddense_attention.py221 class Attention(BaseDenseAttention): class
/external/executorch/examples/mediatek/models/llm_models/
Dmodeling_common.py58 class Attention(nn.Module): class
/external/pytorch/benchmarks/gpt_fast/
Dmodel.py186 class Attention(nn.Module): class
Dmixtral_moe_model.py165 class Attention(nn.Module): class
/external/executorch/examples/models/llama/
Dllama_transformer.py264 class Attention(nn.Module): class
/external/pytorch/torch/testing/_internal/distributed/_tensor/
Dcommon_dtensor.py104 class Attention(nn.Module): class
/external/pytorch/test/inductor/
Dtest_padding.py504 class Attention(nn.Module): class
Dtest_cpu_repro.py3160 class Attention(torch.nn.Module): class