Searched refs:CudnnHandle (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 132 class CudnnHandle { class
136 CudnnHandle(gpu::ScopedActivateExecutorContext context, in CudnnHandle() function in stream_executor::gpu::__anon040dea3a0111::CudnnHandle
183 CudnnHandle GetHandle(GpuExecutor* executor, Stream* stream) { in GetHandle()
190 return CudnnHandle(std::move(context), std::move(lock), handle_); in GetHandle()
993 const CudnnHandle& cudnn, float dropout, uint64 seed, in Create()
1038 const CudnnHandle& cudnn, int input_size, cudnnDataType_t data_type,
1058 CudnnRnnDescriptor(const CudnnHandle& cudnn, gpu::RnnDescriptor rnn_desc, in CudnnRnnDescriptor()
1089 const CudnnHandle& cudnn, int num_layers, int hidden_size, int input_size, in Create()
1285 const CudnnHandle& cudnn, cudnnRNNDescriptor_t rnn_desc, const int layer, in CheckAndFetchProjectionWeights()
1356 const CudnnHandle& cudnn, int input_size, cudnnDataType_t data_type, in Create()
[all …]
|