aten/src/ATen/native/cudnn/RNN.cpp
// TODO: I am not sure if we actually need the 'dropout' and 'train' parameters
// to initialize just the state tensor
I am not sure if we actually need the 'dropout' and 'train' parameters
to initialize just the state tensor
I am not sure if we actually need the ' dropout ' and ' train ' parameters to initialize just the state tensor
Tensor _cudnn_init_dropout_state(const Type& ty, double dropout, bool train, int64_t dropout_seed) {
  auto handle = getCudnnHandle();
  DropoutDescriptor dropout_desc;
  auto dropout_p = train ? dropout : 0;
  dropout_desc.initialize_rng(ty, handle, dropout_p, dropout_seed);
  return dropout_desc.state;
}
Tensor _cudnn_init_dropout_state ( const Type & ty , double dropout , bool train , int64_t dropout_seed ) { auto handle = getCudnnHandle ( ) ; DropoutDescriptor dropout_desc ; auto dropout_p = train ? dropout : 0 ; dropout_desc . initialize_rng ( ty , handle , dropout_p , dropout_seed ) ; return dropout_desc . state ; }
TODO: I am not sure if we actually need the 'dropout' and 'train' parameters
to initialize just the state tensor
TODO : I am not sure if we actually need the ' dropout ' and ' train ' parameters to initialize just the state tensor