Searched refs:CudnnDropoutDescriptor (Results 1 – 1 of 1) sorted by relevance
985 class CudnnDropoutDescriptor { class986 explicit CudnnDropoutDescriptor(DropoutDescriptor handle) in CudnnDropoutDescriptor() function in stream_executor::gpu::__anon040dea3a0311::CudnnDropoutDescriptor990 CudnnDropoutDescriptor(CudnnDropoutDescriptor&&) = default;992 static port::StatusOr<CudnnDropoutDescriptor> Create( in Create()999 return CudnnDropoutDescriptor(std::move(handle)); in Create()1014 return CudnnDropoutDescriptor(std::move(handle)); in Create()1021 SE_DISALLOW_COPY_AND_ASSIGN(CudnnDropoutDescriptor);1066 CudnnDropoutDescriptor dropout_desc, in CudnnRnnDescriptor()1096 CudnnDropoutDescriptor dropout_desc, in Create()1097 CudnnDropoutDescriptor::Create(cudnn, dropout, seed, state_allocator)); in Create()[all …]