Jun Gao dce05da535 fix thrust-transform-reduce error (#5078)

2023-11-21 15:09:35 +08:00

..

csrc

fix thrust-transform-reduce error (#5078)

2023-11-21 15:09:35 +08:00

mha

[npu] add npu support for gemini and zero (#5067)

2023-11-20 16:12:41 +08:00

__init__.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

layer_norm.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

multihead_attention.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

scaled_softmax.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00