tlc-pack / relax

Apache License 2.0
195 stars 58 forks source link

[Op][O2h] Neural network and linear algebra operators #343

Closed MasterJH5574 closed 1 year ago

MasterJH5574 commented 1 year ago

As tracked by #332, this PR is the O2h milestone of the high-level operator introduction plan.

For the first part, following #341, this PR continues to introduce neural network operators. Specifically, this PR introduces

For the second part, this PR introduces the linear algebra operator matmul.

All ops are well-tested.