Commit Graph

8 Commits (6ee8a2e1dbdc35a347677ee15a68963af7731b77)

Author SHA1 Message Date
qingqing01 32d881beab Optimize the rowwise add function.
7 years ago
qiaolongfei 682eee40cb fix math_function warning
7 years ago
Yu Yang 7e214b4985
Speed up ColwiseSum in CPU (#6834)
7 years ago
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
dangqingqing 2673657684 Move RowwiseAdd functor to math_funcion and Add ColwiseSum functor.
8 years ago
dangqingqing 91d4fc6941 Fix compling for softmax_with_cross_entropy_op.
8 years ago
dangqingqing 5f21709938 Fix bug.
8 years ago
dangqingqing f5e367655e Use G++ to compile some cu operators.
8 years ago