Commit Graph

7 Commits (efa50a4525edf786d77568e4b389e37ef180e731)

Author SHA1 Message Date
liaogang b58725bd51 Add cpplint for *.h and cuda *.cu
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 122e83e36c Fix unittest
8 years ago
Yu Yang 1ed237c11e Add skeletons of `mul`, `rowwise_add`, `sigmoid`, `softmax` ops
8 years ago