Commit Graph

12 Commits (6db6475460d45476ec2c8e79ba07b2a0a50608a9)

Author SHA1 Message Date
dangqingqing 0dce16a697 Use bool type for attr in cross_entropy_op.
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