Include protobuf-defined outputs in the graph cutting algorithm (#23557)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/23557
As title states this enables any tensors defined by the user to be outputs, including activations
Reviewed By: yinghai
Differential Revision: D16362993
fbshipit-source-id: b7dc8412c88c46fcf67a3b3953dc4e4c2db8c4aa