search

apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators
https://tvm.apache.org/
Apache License 2.0
11.41k stars 3.4k forks source link

[TOPI] Add dense schedule for fp16 and fp32 using gemm #17091

Open eirenevp opened 3 weeks ago

eirenevp commented 3 weeks ago

Add a new schedule for the dense operator based on the gemm algorithm.

ekalda commented 3 weeks ago

cc @Anndrey24 @lhutton1 @leandron