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