• M
    Mkldnn layout (#11040) · 3ff9ba0e
    mozga-intel 提交于
    * Add MKLDNN layout support in Paddle
    
    Add MKLDNN layout in Paddle so that MKLDNN friendly memory layout
    can be used in MKLDNN enabled OP kernel. Before this commit, NCHW
    is hardcode to be used in all MKLDNN op kernels. As a result,
    non-optimized execution path is selected in MKLDNN primitive which
    bring worse performance.
    Besides framework change, three MKLDNN OP kernels were updated
    for using new MKLDNN layout. They are conv/pool2d/batch_norm.
    Other MKLDNN OP kernels need be also updated in similar way to
    achieve best performance.
    
    * Add MKLDNN layout support in activation OP
    
    * Don't populate layout from input to output when kMKLDNN in
    
    * Refine pool mkldnn op kernel
    
    * MKLDNN layout
    
    * Remove the inferitance from tensor file
    
    * MKLDNN layout: refactoring
    
    * Remove additional #define to register new operator
    
    * Prepare mkldnn tests to work with layout
    3ff9ba0e
activation_op.cc 17.1 KB
反馈
建议
客服 返回
顶部