1. 21 12月, 2018 1 次提交
    • C
      [Feature] Add Temporary Allocator (#14875) · 79bd6dfa
      chengduo 提交于
      * Add Temporal Allocator
      
      * add Temporay Allocator to DeviceContext
      test=develop
      
      * code refine
      test=develop
      
      * fix mean_iou
      test=develop
      
      * Add DeviceTemporaryAllocator
      test=develop
      
      * fix conv_op bug
      test=develop
      
      * small fix
      test=develop
      
      * code refine
      test=develop
      
      * log refine
      test=develop
      
      * fix unit test
      test=develop
      
      * move double check
      
      * refine concat_and_split
      test=develop
      
      * add limit_of_temporary_allocation
      test=develop
      
      * fix name
      test=develop
      79bd6dfa
  2. 14 6月, 2018 1 次提交
    • W
      Add mean IOU op. (#10519) · 6fcdb240
      whs 提交于
      * Add mean_iou op.
      
      * Add unitest for mean iou op.
      
      * Add optional collections of confusion matrix and mean_iou.
      
      * Fix cuda kernel.
      
      * Refine code.
      1. Merge computing in GPU to two kernel.
      2. Use wrong array and correct array instead of confusion matrix.
      
      * Add python api and fix cuda kernel.
      
      * Fix comments.
      
      * Small fix.
      
      * Small fix.
      6fcdb240