• Nick Korovaiko's avatar
    MaxPoolWithIndices (#900) · a174c8c9
    Nick Korovaiko authored
    * MaxPoolWithIndices CPU Fusion
    
    * fix test to pass checks in cpu_fusion
    
    * pass test
    
    * clean up
    
    * add a new pass, add layouts
    
    * remove the opt from cpu_fusion
    
    * refactor cpu_layout logic for maxpool, clean up comments
    
    * add comment w.r.t. indices tensor
    
    * rename to cpu_workspace_insertion
    
    * add CPUWorkspaceInsertion pass for TF
    a174c8c9
cpu_fusion.cpp 50.5 KB