• 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
cpu_fusion.cpp 129 KB