search

d2l-ai / d2l-tvm

Dive into Deep Learning Compiler
https://tvm.d2l.ai
630 stars 97 forks source link

Gpu conv #17

Closed yidawang closed 4 years ago

yidawang commented 4 years ago

Add the conv optimization on GPU. @mli