mindspore2022

30629 Commits

18 Branches

584 MB

243 Download

Tree: 5c72e9d7c6

Commit Graph

Author	SHA1	Message	Date
Xiaoda Zhang	5613c0b974	add a moe implementation: 1) extend the Liner cell for including BatchMatMul implementation, in which the first dimension indicates the expert number; 2) implement a Switch (top1) router; 3) implement a MoE cell, which extends the FeedForward cell.	4 years ago

Author

SHA1

Message

Date

Xiaoda Zhang

5613c0b974

add a moe implementation:

1) extend the Liner cell for including BatchMatMul implementation, in
which the first dimension indicates the expert number;
2) implement a Switch (top1) router;
3) implement a MoE cell, which extends the FeedForward cell.

4 years ago

1 Commits (5c72e9d7c62ea9f61b6a5d12f85fffa4e55711fc)