| D | cuda_dnn.cc | 993 const CudnnHandle& cudnn, float dropout, uint64 seed, in Create() argument
1006 cudnnDropoutGetStatesSize(cudnn.handle(), &state_sizes_in_bytes)); in Create()
1011 handle.get(), cudnn.handle(), dropout, state_memory.opaque(), in Create()
1038 const CudnnHandle& cudnn, int input_size, cudnnDataType_t data_type,
1058 CudnnRnnDescriptor(const CudnnHandle& cudnn, gpu::RnnDescriptor rnn_desc, in CudnnRnnDescriptor() argument
1089 const CudnnHandle& cudnn, int num_layers, int hidden_size, int input_size, in Create() argument
1097 CudnnDropoutDescriptor::Create(cudnn, dropout, seed, state_allocator)); in Create()
1151 cudnn.handle(), /*rnnDesc=*/rnn_desc.get(), in Create()
1160 cudnn.handle(), /*rnnDesc=*/rnn_desc.get(), in Create()
1191 cudnn, input_size, data_type, rnn_desc.get(), in Create()
[all …]
|