| 29 | }; |
| 30 | |
| 31 | size_t DropoutForwardImpl::get_mask_size_in_bytes(const TensorLayout& inp) { |
| 32 | if (inp.is_empty()) { |
| 33 | return 0; |
| 34 | } |
| 35 | |
| 36 | size_t reserve_space_size_in_bytes = 0; |
| 37 | DropoutTensorDesc ddesc(inp); |
| 38 | cudnn_check( |
| 39 | cudnnDropoutGetReserveSpaceSize(ddesc.desc, &reserve_space_size_in_bytes)); |
| 40 | return reserve_space_size_in_bytes; |
| 41 | } |
| 42 | |
| 43 | void DropoutForwardImpl::exec( |
| 44 | _megdnn_tensor_in inp, _megdnn_tensor_out oup, _megdnn_tensor_out mask, |