• Chris Sullivan's avatar
    Add MaxPool as an explicit argument to MaxPoolBackprop (#2065) · 67c0488b
    Chris Sullivan authored
    * remove forward op
    
    * fix bbrks
    
    * fix pybind c-tor for max_pool_bprop
    
    * Add new c-tor to MaxPoolBackprop that takes MaxPool as an explicit argument. Add serializer support.
    
    * Add nvgpu support for new backward pooling c-tor, and calculate fprop when it isn't available.
    
    * Add extra layout for 3 arg maxpool backprop.
    
    * Formatting.
    
    * cpu_workspace_insertion to expect 3-arg maxpool bprop
    
    * GPU: add bprop_needs_pooling flag to primitive hash
    
    * Update INTELGPU arguments_check for MaxPoolBackprop and GPU invocation for avg pool.
    67c0488b
Name
Last commit
Last update
.ci Loading commit data...
cmake Loading commit data...
contrib/docker Loading commit data...
doc Loading commit data...
licenses Loading commit data...
maint Loading commit data...
python Loading commit data...
src Loading commit data...
test Loading commit data...
.clang-format Loading commit data...
.gitattributes Loading commit data...
.gitignore Loading commit data...
.gitmodules Loading commit data...
.travis.yml Loading commit data...
ABOUT.md Loading commit data...
CMakeLists.txt Loading commit data...
CODEOWNERS Loading commit data...
CONTRIB.md Loading commit data...
FAQs.md Loading commit data...
INSTALL.md Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
VERSION.in Loading commit data...
changes.md Loading commit data...