• T
    MKLDNN elementwis_add with default broadcast operations (#11544) · e26f51ce
    Tomasz Patejko 提交于
    * elementwise_add with bcast: Brian's implementation by Brian added, with default bcasts
    
    * elementwise_add with bcast: GetExpectedKernelType added to elementwise_op
    
    * elementwise_add with bcast: use_mkldnn attribute added
    
    * elementwise_add with bcast: changes after review and some formatting
    
    * elementwise_add with bcast: changes after style check
    
    * elementwise_add with bcast: changes after style check cont.
    
    * elementwise_add with bcast: MKLDNN unittests added
    
    * elementwise_add with bcast: original unittests with use_mkldnn flag
    
    * elementwise_add with bcast: handling of MKLDNN format corrected
    
    * elementwise_add with bcast: setting MKLDNN format turned into lambda
    
    * elementwise_add with bcast: MKDNN format setting turned into separate function
    
    * elementwise_add with bcast: condition for choosing MKLDNN simplified
    
    * elementwise_add with bcast: fix for MKLDNN format set incorrectly in bcasts
    
    * elementwise_add with bcast: changes in unittests for broadcasts
    
    * elementwise_add with bcast: fixes in unittests regarding dimensions
    
    * elementwise_add with bcast: bring back correct format setting in mklml grad path
    
    * elementwise_add with bcast: fixed compilation error
    e26f51ce
test_elementwise_add_op.py 8.9 KB