• 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
..
cpu_assignment.cpp Loading commit data...
cpu_assignment.hpp Loading commit data...
cpu_collapse_dims.cpp Loading commit data...
cpu_collapse_dims.hpp Loading commit data...
cpu_concat_inputs.cpp Loading commit data...
cpu_concat_inputs.hpp Loading commit data...
cpu_fusion.cpp Loading commit data...
cpu_fusion.hpp Loading commit data...
cpu_horizontal_fusion.cpp Loading commit data...
cpu_horizontal_fusion.hpp Loading commit data...
cpu_layout.cpp Loading commit data...
cpu_layout.hpp Loading commit data...
cpu_loop_kernel_fusion.cpp Loading commit data...
cpu_loop_kernel_fusion.hpp Loading commit data...
cpu_mat_fusion.cpp Loading commit data...
cpu_mat_fusion.hpp Loading commit data...
cpu_memory_optimization.cpp Loading commit data...
cpu_memory_optimization.hpp Loading commit data...
cpu_post_layout_optimizations.cpp Loading commit data...
cpu_post_layout_optimizations.hpp Loading commit data...
cpu_reshape_sinking.cpp Loading commit data...
cpu_reshape_sinking.hpp Loading commit data...
cpu_rnn_fusion.cpp Loading commit data...
cpu_rnn_fusion.hpp Loading commit data...
cpu_workspace_insertion.cpp Loading commit data...
cpu_workspace_insertion.hpp Loading commit data...
halide_subgraph_extraction.cpp Loading commit data...
halide_subgraph_extraction.hpp Loading commit data...