search

liuliu / ccv

C-based/Cached/Core Computer Vision Library, A Modern Computer Vision Library
http://libccv.org
Other
7.1k stars 1.71k forks source link

computation graph optimization of pytorch #220

Open knsong opened 4 years ago

knsong commented 4 years ago

hi, thank you for the great work, But does it support computation graph optimization for pytorch?

liuliu commented 4 years ago

Not at the moment. I should add ONNX support at some point once everything is more fleshed out. The optimization performed at symbolic graph level. For model import / export, it likely happens at somewhere between symbolic graph + tensor arena allocation and the concrete graph. The API haven't thought out yet.