1. 23 5月, 2018 2 次提交
    • F
      expose random_data_generator · c2436f2c
      fengjiayi 提交于
      c2436f2c
    • W
      Enhance reduce op (#10708) · 8655904b
      whs 提交于
      * Enhance reduce op for multi dims.
      
      * Uncomment some unitest.
      
      * Uncomment unitest.
      
      * Remove unused code.
      
      * Fix infershape and python wrapper.
      
      * Add more examples.
      
      * Fix l2_normalize.
      
      * Fix normalization_wrapper.
      
      * Polish code.
      1. Rename unitest function.
      2. Rename const variable.
      8655904b
  2. 22 5月, 2018 4 次提交
  3. 21 5月, 2018 1 次提交
  4. 20 5月, 2018 1 次提交
  5. 17 5月, 2018 2 次提交
  6. 11 5月, 2018 1 次提交
    • F
      trainer.test() (#10453) · ba57348f
      fengjiayi 提交于
      * a draft of trainer.test()
      
      * polish trainer.test()
      
      * polish trainer.test()
      
      * update code format
      
      * update
      
      * polish code
      
      * polish code
      
      * polish code
      
      * Make trainer.test follow the rule of returning [loss, metric, metric, ..]
      ba57348f
  7. 10 5月, 2018 1 次提交
  8. 04 5月, 2018 1 次提交
  9. 03 5月, 2018 1 次提交
    • T
      MKLDNN implementation of batch normalization (#9904) · 4a497b82
      Tomasz Patejko 提交于
      * Initial implementation of forward pass for MKLDNN batch norm
      
      * Added attributes for MKLDNN batch norm
      
      * MKLDNN batch norm forward pass passes unittest. Started working on backward
      
      * Backward pass for MKLDNN batch norm added
      
      * MKLDNN batch norm: scoring added to forward pass
      
      * MKLDNN batch norm: bias as input added; handling AnyLayout when kernel is looked up
      
      * MKLDNN batch norm: python unit tests added; mkldnn tests removed
      
      * MKLDNN batch norm: changes required by cpplint
      
      * MKLDNN batch norm: refactoring the operator
      
      * MKLDNN batch norm: saved variance inversed in backward pass for correct execution of MKLDNN unit tests
      
      * MKLDNN batch norm: refctoring, function for static/const cast to void* added
      
      * MKLDNN batch norm: remove AnyLayout from batch norm
      
      *  MKLDNN batch norm: only NCHW format is supported. Unittests refactored
      
      * MKDNN batch norm: use_mkldnn added to attributes
      
      * MKLDNN batch norm: AnyLayout removed from unittest
      
      * MKLDNN batch norm: added CUDNN defines to batch norm
      
      * MKLDNN batch norm: undefined data_format variable corrected
      
      * MKLDNN batch norm: use_cudnn added, use of setUp method for configuring attributes
      
      * MKLDNN batch norm: added use_cudnn attribute to batch norm operator
      
      * MKLDNN batch norm: correcting batch norm unit tests for MKLDNN
      
      * MKLDNN batch norm: MKLDNN tests moved to another file; reverting changes for saved variance not being inverted
      
      * Change default layout to NCHW
      
      * MKLDNN batch norm: init_kernel_type method added to unit tests
      
      * MKLDNN batch norm: style changes
      
      * MKLDNN batch norm: unit tests refactored
      
      * MKLDNN batch norm: added use_mkldnn attribute to batch norm python interface
      4a497b82
  10. 02 5月, 2018 2 次提交
  11. 27 4月, 2018 5 次提交
  12. 26 4月, 2018 1 次提交
  13. 24 4月, 2018 1 次提交
  14. 22 4月, 2018 1 次提交
  15. 18 4月, 2018 3 次提交
  16. 17 4月, 2018 3 次提交
  17. 16 4月, 2018 3 次提交
  18. 12 4月, 2018 1 次提交
    • Q
      Dist transpiler support prefetch (#9714) · 4c55a602
      Qiao Longfei 提交于
      * init
      
      * add some check
      
      * add dist transpile logic
      
      * add insert op for block
      
      * init change get_pserver_program
      
      * optimize code
      
      * fix a bug
      
      * can run now
      
      * start to do table split
      
      * start to process table gradient
      
      * complete pserver part
      
      * can send_vars now
      
      * revert cpplint
      
      * fix a bug
      
      * optimize code
      
      * move dist test to models
      
      * revert the interface of distribute_transpiler.transpile
      
      * fix prefetch_block
      
      * optimize trainspiler code
      
      * add comment to sum_op
      
      * add warning log
      
      * fix comment
      
      * fix test_send_recv
      
      * fix test_send_recv
      
      * fix train with no distributed table
      
      * optimize GetDims
      4c55a602
  19. 11 4月, 2018 3 次提交
  20. 10 4月, 2018 2 次提交
  21. 09 4月, 2018 1 次提交