1. 16 11月, 2018 1 次提交
    • T
      MKLDNN residual connections fuse pass: · 7423748e
      Tomasz Patejko 提交于
      * implements reachability check between identity node and non-identity argument to elementwise_add
      * implements handling identity node as x and as y argument to elementwise_add
      7423748e
  2. 15 11月, 2018 1 次提交
    • S
      add mkldnn prop_kind phase for inference-only case to pooling and activations (#14278) · 8a1eeec5
      Sylwester Fraczek 提交于
      * add is_test to pooling and activations
      
      add prop_kind support for layers activation. conv and pooling
      
      add a pass that sets is_test to true
      
      add transpiler version of is_test pass
      
      test=develop
      
      * patch test and pass
      
      test=develop
      
      * add pass to analyzer.h
      
      test=develop
      
      * add is_test attr description & pass only on mkldnn
      
      in:
      activation_op.cc
      batch_norm_op.cc
      conv_op.cc
      dropout_op.cc
      lrn_op.cc
      pool_op.cc
      sequence_pool_op.cc
      softmax_op.cc
      
      * fix is_test handling for activation pool and conv
      
      * change description of is_test for all layers again
      
      * remove GetAttr(use_mkldnn) from pass
      
      * rename correct_mkldnn_test_phase to is_test
      
      and remove dependency on MKLDNN
      test=develop
      
      * review fix magic number
      
      * two if(..)s into one
      
      * Check is_test once and pass mkldnn forward prop kind
      
      * dereference shared_ptr with * (without get())
      
      test=develop
      
      * add is_test_pass back
      
      test=develop
      8a1eeec5
  3. 14 11月, 2018 6 次提交
  4. 12 11月, 2018 3 次提交
  5. 09 11月, 2018 2 次提交
  6. 08 11月, 2018 3 次提交
  7. 07 11月, 2018 2 次提交
  8. 06 11月, 2018 6 次提交
  9. 05 11月, 2018 2 次提交
  10. 02 11月, 2018 4 次提交
  11. 31 10月, 2018 1 次提交
    • S
      add depthwise conv mkldnn pass · 4e2aaf01
      Sylwester Fraczek 提交于
      added depthwise conv mkldnn pass which for MKLDNN changes depthwise_conv operator to conv operator because for mkldnn this is the same api
      test=develop
      4e2aaf01
  12. 30 10月, 2018 2 次提交
  13. 29 10月, 2018 1 次提交
    • W
      [1.1] [project] train imagenet using large batch size (#13766) · 26200f2e
      Wu Yi 提交于
      * fix nccl2 lars dist support
      
      * put lars in momentum op
      
      * add tests lars
      
      * fix ci
      
      * fix cpu kernel
      
      * soft warning
      
      * remove lars in test_recognize_digits.py
      
      * move to another op
      
      * add file
      
      * update api.spec test=develop
      
      * update test=develop
      
      * fix api.spec test=develop
      
      * wip
      
      * wip, finish grad merge ops
      
      * wip, finish graph build
      
      * wip test running
      
      * work on 1 gpu
      
      * workable version
      
      * update
      
      * fix tests
      
      * fuse broadcast op
      
      * fix compile failed
      
      * refine
      
      * add batch merge test mnist
      
      * fix CI test=develop
      
      * fix build
      
      * use independent bn params for batch merge test=develop
      
      * update api.spec
      
      * follow comments and for test
      
      * wip
      
      * refine tests test=develop
      
      * follow comments test=develop
      
      * remove startup bn modify test=develop
      
      * follow comments test=develop
      
      * fix merge test=develop
      26200f2e
  14. 28 10月, 2018 1 次提交
    • W
      [1.1] fix graph num hang (#14072) · 9da9b192
      Wu Yi 提交于
      * fix graph num hang test=develop
      
      * re-enable tests test=develop
      
      * re-enable graph num check test=develop
      
      * fix multi device pass role check test=develop
      9da9b192
  15. 26 10月, 2018 1 次提交
  16. 25 10月, 2018 1 次提交
  17. 23 10月, 2018 2 次提交
  18. 21 10月, 2018 1 次提交