pytorch
a6fa6a6c - [fx minimizer] Add an option to minimizer to allow return all intermediate results (#57279)

Commit
3 years ago
[fx minimizer] Add an option to minimizer to allow return all intermediate results (#57279) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57279 Added an option "return_intermediate". If true, when building the submodule we want to run , we will replace the output with all the nodes, so that intermediate results of all the nodes will be returned as output. This is recommended to use with `run_node()` function. Test Plan: `buck test glow/fb/nnpi/lowering:net_min_tests` Reviewed By: khabinov Differential Revision: D27913887 fbshipit-source-id: 5a3eab02da05214fb9adeb25656c267b58075b1d
Author
Parents
Loading