Commit Graph

13 Commits (578d60bf4adf8659a28d9157fa5c5511b17900b1)

Author SHA1 Message Date
chengduo 95862a547c
Merge pull request #7014 from chengduoZH/profiling/refine_drop_out
7 years ago
Luo Tao 761b329793 unify the indentation of license
7 years ago
chengduoZH 52119d62a7 refine
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
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