Stella Laurenzo 0c73c535d6 Capture backward conv and copy_ kernels. ... * This is sufficient to capture the forward and backward pass and gradients of a convolutional model with an nllloss. * As with the forward conv, the backward conv is a special case wrapped in an enigma on the PyTorch side. There aren't many like it, so special casing is just what we do. * When I traced this, I found that the copy_ op is not yet boxing compatible so I had to map it manually. If there are many more like this, I'll probably do something a bit more clever to reduce duplication. * This exposes new signature patterns that will need to be handled by the ATen lowering. Will take care of that next: It will be nice to have an e2e of a non-trivial case with full gradients. * Fixes #97.		2020-10-30 22:59:26 -07:00
..
c10_dispatch	Capture backward conv and copy_ kernels.	2020-10-30 22:59:26 -07:00
type_dispatch	Add capture function arguments.	2020-10-01 18:59:58 -07:00
CMakeLists.txt	Start reworking towards a shared library build.	2020-10-09 16:02:58 -07:00
init_python_bindings.cpp	NFC: Transition ATen passes to tablegen registration.	2020-10-22 17:12:44 -07:00
init_python_bindings.h	Add capture function arguments.	2020-10-01 18:59:58 -07:00
pybind.h	Add capture function arguments.	2020-10-01 18:59:58 -07:00