1. 02 7月, 2018 1 次提交
  2. 26 6月, 2018 1 次提交
    • 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