dropout_op_kernel.cc
3.07 KB
-
Fix the bug on dropout cuda kernel · b37d4e5eSummary: We forgot to handle the inplace case that generated the random elements on the output(i.e. the input). Besides, this commit also fixes the omitted `RunOnDevice` for cudnn activations, which will rightly dispatches the implementation.
Ting PAN committed