Will Feng b5d15315d8 Improve C++ maxpool and avgpool (#26521) ... Summary: This PR makes the following improvements: 1. Add `forward_with_indices` method to all C++ MaxPool modules, to return the max indices along with the outputs. (We can't make two `forward` methods that return different types based on input, because that will break the type deduction of `torch::detail::return_type_of_forward_t`) 2. Add `max_poolNd_with_indices` to `torch::nn::functional`, to be used when indices of the max values are needed. (We can't merge this with `torch::nn::functional::max_poolNd` because the return type of `max_poolNd` has to be defined statically). 3. Improve `pretty_print` of C++ MaxPoolNd and AvgPoolNd modules to match the Python `extra_repr`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/26521 Differential Revision: D17507358 Pulled By: yf225 fbshipit-source-id: b6c0e2b27b38378cdc0c75f4bfc797b3c6b17cd9		2019-09-25 13:52:58 -07:00
..
api	Improve C++ maxpool and avgpool (#26521)	2019-09-25 13:52:58 -07:00
common	Trim libshm deps, move tempfile.h to c10 (#17019)	2019-02-13 19:38:35 -08:00
dist_autograd	Attach 'send' autograd function to the autograd graph as part of RPC. (#24876)	2019-09-01 23:54:01 -07:00
jit	autodiff changes to enable profiling	2019-09-25 10:11:44 -07:00
__init__.py	Add train() / eval() / is_training() to C++ ScriptModule API (#16044)	2019-02-01 13:07:38 -08:00