• 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
max_pool.cpp 2.31 KB