Commit Graph

22 Commits (097f533bcabd80876a3a03b838f7b4ed481eb5c5)

Author SHA1 Message Date
qijun b611a479fc fix gpu build error
7 years ago
qijun 79def5e634 refine CrossEntropyFunctor
7 years ago
Yu Yang 3a5693e0a8 Add Skeleton of Double support
7 years ago
caoying03 97509b68e7 cross entropy as a functor to avoid duplicated codes.
7 years ago
caoying03 000d75116f fix backward op.
7 years ago
caoying03 201c2bcf20 delete redundant codes.
7 years ago
caoying03 6735585b0f fix cpu kernel with soft labels.
7 years ago
caoying03 30bfaab36e Merge branch 'develop' into optimize_cross_entropy_kernel
7 years ago
caoying03 141b8dbce6 update the backward kernel.
7 years ago
dangqingqing 0dce16a697 Use bool type for attr in cross_entropy_op.
8 years ago
caoying03 a3a8a0900d optimize cross entropy kernel by using reduce.
8 years ago
Xinghai Sun d8046da0cd Use soft_label attribute for cross-entropy.
8 years ago
Xinghai Sun e87068290e Update cross entropy operator by following reviewer's comments.
8 years ago
Xinghai Sun 6d60352e7e Add soft-label support for cross-entropy operator.
8 years ago
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