Index of
/ipfs/bafybeihgzxfl2xxbn6naq7cldwshxpnihu7ffuearjy5tjf5z43a5z2sv4/deep_astro_uda/apex/contrib/multihead_attn
bafybeiekpxzhmcpfc2v3g4vagaisi5zy3xjt3nwqdohtsoduolrgkrkfym
244 kB
|
|
.. | ||
|
|
MHA_bwd.png | bafk…y45q | 87 kB |
|
|
MHA_fwd.png | bafk…kqt4 | 84 kB |
|
|
README.md | bafk…d2am | 2.3 kB |
|
|
__init__.py | bafk…ydii | 176 B |
|
|
encdec_multihead_attn.py | bafk…fwkq | 7.7 kB |
|
|
encdec_multihead_attn_func.py | bafk…bq34 | 17 kB |
|
|
fast_encdec_multihead_attn_func.py | bafk…gazy | 3.0 kB |
|
|
fast_encdec_multihead_attn_norm_add_func.py | bafk…7oby | 4.3 kB |
|
|
fast_self_multihead_attn_func.py | bafk…f5qy | 7.7 kB |
|
|
fast_self_multihead_attn_norm_add_func.py | bafk…fobi | 3.5 kB |
|
|
mask_softmax_dropout_func.py | bafk…y6vq | 2.5 kB |
|
|
self_multihead_attn.py | bafk…64pu | 10 kB |
|
|
self_multihead_attn_func.py | bafk…td2e | 14 kB |
