Commit Graph

8 Commits (bc6bae8d533042c13e7cce9640a9f0857e851c94)

Author SHA1 Message Date
dangqingqing 8f6c8780a5 Replace functor by function.
8 years ago
dangqingqing 26475cd9ba Use clipping log in cuda kernel, making it same with CPU.
8 years ago
dangqingqing 49aa2c042c Implement GPU kernel for cross entropy operator.
8 years ago
dongzhihong 9e25988f73 "net op alias"
8 years ago
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
Qiao Longfei 2685765905 add cross-entropy-op (#2965)
8 years ago