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