Commit Graph

14 Commits (f086ebb8b9b8e21f7d0d2056f6a961da080edfb2)

Author SHA1 Message Date
Luo Tao 761b329793 unify the indentation of license
7 years ago
Yibing Liu c2b1ddb6a8 Correct the dropout_op's computation in test
7 years ago
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
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