Searched refs:dropout_desc (Results 1 – 1 of 1) sorted by relevance
1066 CudnnDropoutDescriptor dropout_desc, in CudnnRnnDescriptor() argument1082 dropout_desc_(std::move(dropout_desc)), in CudnnRnnDescriptor()1096 CudnnDropoutDescriptor dropout_desc, in Create()1147 /*dropoutDesc=*/dropout_desc.handle(), in Create()1153 /*dropoutDesc=*/dropout_desc.handle(), /*inputMode=*/input_mode, in Create()1198 std::move(dropout_desc), std::move(params_desc)); in Create()1291 cudnnDropoutDescriptor_t dropout_desc; in CheckAndFetchProjectionWeights() local1302 /*dropoutDesc=*/&dropout_desc, in CheckAndFetchProjectionWeights()1313 /*dropoutDesc=*/&dropout_desc, in CheckAndFetchProjectionWeights()