mirror of https://github.com/hpcaitech/ColossalAI
dce05da535 | ||
---|---|---|
.. | ||
csrc | ||
mha | ||
__init__.py | ||
layer_norm.py | ||
multihead_attention.py | ||
scaled_softmax.py |
dce05da535 | ||
---|---|---|
.. | ||
csrc | ||
mha | ||
__init__.py | ||
layer_norm.py | ||
multihead_attention.py | ||
scaled_softmax.py |