Searched refs:dropout_desc (Results 1 – 1 of 1) sorted by relevance
1011 CudnnDropoutDescriptor dropout_desc, in CudnnRnnDescriptor() argument1026 dropout_desc_(std::move(dropout_desc)), in CudnnRnnDescriptor()1040 CudnnDropoutDescriptor dropout_desc, in Create()1049 /*numLayers=*/num_layers, /*dropoutDesc=*/dropout_desc.handle(), in Create()1104 std::move(dropout_desc), std::move(params_desc)); in Create()