1. 12 10月, 2017 3 次提交
    • Cudnn conv op (#4195) · a3ccbdb3
      武毅 提交于
      * add cudnn_conv_op
      
      * WIP
      
      * update
      
      * update
      
      * fix grad check
      
      * use platform::memory
      
      * add support group for cudnn
      
      * update
      
      * follow comments
      
      * fix onlycpu build
      
      * update cuda define
      
      * follow comments
      
      * follow comments
      
      * merge with updates
      
      * fix compile error
      
      * follow comments
      
      * follow comments
      a3ccbdb3
    • A
      Adding the Thresholded Relu Op (#4685) · b504a234
      Abhinav Arora 提交于
      * Adding thresholded_relu op
      * Adding test for thresholded relu op
      b504a234
    • Q
      Unify CUDA stream in Tensor CopyFrom interface (#4692) · 2603cb7e
      QI JUN 提交于
      * init
      
      * unify CopyFrom interface
      
      * fix gpu build error
      
      * fix bug in tensor_py.h
      
      * refine code comments and add TODO list
      
      * fix conflicts in FeedOp and FetchOp
      2603cb7e
  2. 11 10月, 2017 33 次提交
  3. 10 10月, 2017 4 次提交