Commit Graph

10 Commits (58561d8f56b9f5436a12f42994ece7481bde8f9d)

Author SHA1 Message Date
liaogang c6186120c3 fix softmax_op code line > 80
8 years ago
liaogang 430b421906 Fix conflicts
8 years ago
liaogang b58725bd51 Add cpplint for *.h and cuda *.cu
8 years ago
Qiao Longfei d953611e86 Softmax grad op (#3164)
8 years ago
qijun 61f94f0002 add EIGEN_USE_GPU macro to op.cu file
8 years ago
Yu Yang efc119b43b Add type_alias to import framework into ops
8 years ago
qijun 736d078cbf replace Tensor::tensor to EigenTensor::From
8 years ago
qijun b6c075527c implement some basic OpKernel
8 years ago
Yu Yang 5847b96a61 Follow comments, correct implement `DDim::size()`
8 years ago
Yu Yang 1ed237c11e Add skeletons of `mul`, `rowwise_add`, `sigmoid`, `softmax` ops
8 years ago