• Jayaram Bobba's avatar
    IAT: More convolution folding optimizations (#1712) · 00b4453d
    Jayaram Bobba authored
    * Check output shape when setting memory layout for slice op.
    
    * Miscellaneous fusion and other optimizations for inception-resnetv2
    - ConvBias Batchnorm folding
    - ConvBias Affine folding
    - Check if MKLDNN can slice a given layout and select layouts
      appropriately
    
    * Fixed unit test and bug in conv bias pattern
    
    * Addressed PR feedback
    
    * Addressed PR feedback
    00b4453d
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_post_layout_optimizations.cpp Loading commit data...
cpu_post_layout_optimizations.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...