Searched defs:DropoutDescriptor (Results 1 – 2 of 2) sorted by relevance
227 struct TORCH_CUDA_CPP_API DropoutDescriptor struct243 …AT_CUDNN_CHECK(cudnnSetDropoutDescriptor(mut_desc(), handle, dropout, state.data_ptr(), state_size… in initialize_rng() argument270 DropoutDescriptor dropout_desc_; argument
538 using DropoutDescriptor = typedef