Commit Graph

8 Commits (fa54d9d8c03e4061528c4e3ad154b806cd0c35e7)

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