Searched refs:CudnnDropoutDescriptor (Results 1 – 1 of 1) sorted by relevance
926 class CudnnDropoutDescriptor { class927 explicit CudnnDropoutDescriptor(DropoutDescriptor handle) in CudnnDropoutDescriptor() function in stream_executor::gpu::__anonf41f84630311::CudnnDropoutDescriptor931 CudnnDropoutDescriptor(CudnnDropoutDescriptor&&) = default;933 static port::StatusOr<CudnnDropoutDescriptor> Create( in Create()940 return CudnnDropoutDescriptor(std::move(handle)); in Create()955 return CudnnDropoutDescriptor(std::move(handle)); in Create()962 SE_DISALLOW_COPY_AND_ASSIGN(CudnnDropoutDescriptor);1011 CudnnDropoutDescriptor dropout_desc, in CudnnRnnDescriptor()1040 CudnnDropoutDescriptor dropout_desc, in Create()1041 CudnnDropoutDescriptor::Create(cudnn, dropout, seed, state_allocator)); in Create()[all …]