Commit Graph

30 Commits (f086ebb8b9b8e21f7d0d2056f6a961da080edfb2)

Author SHA1 Message Date
Luo Tao 761b329793 unify the indentation of license
7 years ago
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
dangqingqing e9082bb78e Resume unit testing.
7 years ago
typhoonzero 272f3e6d43 refine get cuda context
7 years ago
QI JUN 008f40ce09
support sparse output for lookup table grad op (#5145)
7 years ago
Yu Yang aa3de3571d Polish unit test for xe, generate probablities (#5096)
7 years ago
Yu Yang 75d0c79015 Change Name convention of operator attributes (#4807)
7 years ago
qijun 5be10872f9 add selected_rows add cpu functor
7 years ago
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.
7 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