2 Commits

Author	SHA1	Message	Date
Yiming Wu	d1b920b44f	keep net type info when generating model complete net (#11032) ... Summary: keep net type info when generating model complete net. This will keep the performance optimization option Pull Request resolved: https://github.com/pytorch/pytorch/pull/11032 Reviewed By: wat3rBro Differential Revision: D9564125 Pulled By: harouwu fbshipit-source-id: c6546af9b1d4ff5eddf6124e24a5da1b8baf47df	2018-09-04 21:10:06 -07:00
Yiming Wu	579962f2a8	reroute tensor feature in core.Net and generate one net feature in model_helper (#10528) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/10528 adding 2 features to core and model_helper - reroute_tensor which supports op insertion on net level - model_helper complete net and cut net used for full graph analysis Differential Revision: D9330345 fbshipit-source-id: 56341d3f500e72069ee306e20266c8590ae7985a	2018-08-15 16:40:15 -07:00