Commit Graph

9 Commits (f1a9efcac5743e05fc4e1dbc53c08f69ec19d5a8)

Author SHA1 Message Date
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
QI JUN 3a76062c84
support testing when training and handle dropout and batch_norm operator in testing mode (#5734)
7 years ago
Yu Yang 3a5693e0a8 Add Skeleton of Double support
8 years ago
dangqingqing 72ba02701b Add bool type for attribute and use it in dropout_op.
8 years ago
Xinghai Sun ffeeef82f3 Remove unnecessary mask operations in test phase for dropout operator.
8 years ago
Xinghai Sun 585d12a307 Add is_training attr and testing phrase compuation to dropout operator.
8 years ago
Xinghai Sun 32645b5231 Move dropout gpu kernel to dropout_op.cu.
8 years ago
Xinghai Sun b1a185524f Fixed SEGFAULT of dropout operator in GPU.
8 years ago
Xinghai Sun 9a44f3d6da Add dropout operator.
8 years ago