apache / tvm

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

[Relax][PyTorch] Support neural network ops for ExportedProgram importer #17426

Closed mshr-h closed 2 months ago

mshr-h commented 2 months ago

Part of #17346 cc @tqchen @yongwww @vinx13