pytorch
230f7f9b - Include protobuf-defined outputs in the graph cutting algorithm (#23557)

Commit
5 years ago
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
Author
Parents
Loading