Searched refs:CudnnHandle (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 133 class CudnnHandle { class
137 CudnnHandle(gpu::ScopedActivateExecutorContext context, mutex_lock lock, in CudnnHandle() function in stream_executor::gpu::__anonf41f84630111::CudnnHandle
184 CudnnHandle GetHandle(GpuExecutor* executor, Stream* stream) { in GetHandle()
190 return CudnnHandle(std::move(context), std::move(lock), handle_); in GetHandle()
934 const CudnnHandle& cudnn, float dropout, uint64 seed, in Create()
979 const CudnnHandle& cudnn, int input_size, cudnnDataType_t data_type,
1003 CudnnRnnDescriptor(const CudnnHandle& cudnn, gpu::RnnDescriptor rnn_desc, in CudnnRnnDescriptor()
1033 const CudnnHandle& cudnn, int num_layers, int hidden_size, int input_size, in Create()
1157 const CudnnHandle& cudnn, int input_size, cudnnDataType_t data_type, in Create()
1470 const CudnnHandle& cudnn, const CudnnRnnDescriptor& rnn_desc, in CheckRNNParameterSize()
[all …]
|