• Jayaram Bobba's avatar
    Switch to using more expressive layout descriptors instead of numeric layout names (#1278) · 69c51c27
    Jayaram Bobba authored
    * Switch to using mkldnn memory descriptors for layout
    
    * More changes for using mkldnn descriptor instead of format
    
    * Removed mkldnn format from cpu layout descriptor. TODO - shuffle folding
    
    * Rotate mkldnn layouts on transpose
    
    * Modifications to builder reshape to skip rotated layouts
    
    * More fixes to layouts and removes axis order from cpu layout descriptor
    
    * Code cleanup
    
    * Removed shuffle folding pass since the functionality is subsumed by the layout pass
    
    * Canonicalize a few more formats to keep MKLDNN happy.
    
    * Style fixes
    
    * Style fixes
    
    * Style fixes
    
    * Addressed PR feedback and added reshape passthrough for non-transpose cases
    
    * Adjust named formats for weights tensors to keep MKLDNN happy
    
    * Style fixes
    
    * resolved merge issues
    69c51c27
cpu_fusion.cpp 116 KB