Commit Graph

11 Commits (a061a85eafa4d50f8d8e4e8b2e08bc75005f82f1)

Author SHA1 Message Date
QI JUN 3a76062c84
support testing when training and handle dropout and batch_norm operator in testing mode (#5734)
7 years ago
Qiao Longfei fc68290bcc update _create_op_func_ and support generate dropout layer (#5134)
7 years ago
Yu Yang 3a5693e0a8 Add Skeleton of Double support
7 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 0532662902 Merge branch 'develop' into dropout
8 years ago
Xinghai Sun 963a4f3c4e Update by following reviewers' comments.
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