computation graph optimization of pytorch

[knsong]

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

[liuliu]

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.