• 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
..
onnx_import Loading commit data...
ops Loading commit data...
passes Loading commit data...
runtime Loading commit data...
types Loading commit data...
axis_set.cpp Loading commit data...
axis_set.hpp Loading commit data...
axis_vector.cpp Loading commit data...
axis_vector.hpp Loading commit data...
coordinate.cpp Loading commit data...
coordinate.hpp Loading commit data...
coordinate_diff.cpp Loading commit data...
coordinate_diff.hpp Loading commit data...
function.cpp Loading commit data...
function.hpp Loading commit data...
node.cpp Loading commit data...
node.hpp Loading commit data...
node_vector.cpp Loading commit data...
node_vector.hpp Loading commit data...
parameter_vector.cpp Loading commit data...
parameter_vector.hpp Loading commit data...
pyngraph.cpp Loading commit data...
pyngraph_onnx_import.cpp Loading commit data...
serializer.cpp Loading commit data...
serializer.hpp Loading commit data...
shape.cpp Loading commit data...
shape.hpp Loading commit data...
strides.cpp Loading commit data...
strides.hpp Loading commit data...
util.cpp Loading commit data...
util.hpp Loading commit data...
util.py Loading commit data...