• Adam Procter's avatar
    Partial Shapes and Types, Part 4m: BatchNorm and backprops (#1904) · 722903ac
    Adam Procter authored
    * Implement partial shape/type propagation for Convolution; fail for want of unit tests
    
    * Implement unit tests for partial shapes/types for Convolution
    
    * Implement partial shape/type validation for BatchNormInference, BatchNormTraining, BatchNormTrainingBackprop with unit tests
    
    * Formatting
    
    * Update CPU and GPU backends to deal with elimination of BatchNormBase
    
    * Update BatchNormTrainingWithStats to use templated emit_BatchNorm function
    
    * Restore five-argument BatchNormTraining ctor for now; #1901 will eliminate it
    
    * Replace enum for input indices with symbolic constants
    
    * Update intelgpu/visualize_tree.cpp to deal with removal of BatchNormBase
    722903ac
type_prop.cpp 518 KB