Searched refs:CudnnHandle (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 160 class CudnnHandle { class
164 CudnnHandle(gpu::ScopedActivateExecutorContext context, in CudnnHandle() function in stream_executor::gpu::__anon467f18dd0111::CudnnHandle
211 CudnnHandle GetHandle(GpuExecutor* executor, Stream* stream) { in GetHandle()
221 return CudnnHandle(std::move(context), std::move(lock), handle_); in GetHandle()
1167 const CudnnHandle& cudnn, float dropout, uint64_t seed, in Create()
1213 const CudnnHandle& cudnn, int input_size, cudnnDataType_t data_type,
1233 CudnnRnnDescriptor(const CudnnHandle& cudnn, gpu::RnnDescriptor rnn_desc, in CudnnRnnDescriptor()
1264 const CudnnHandle& cudnn, int num_layers, int hidden_size, int input_size, in Create()
1463 const CudnnHandle& cudnn, cudnnRNNDescriptor_t rnn_desc, const int layer, in CheckAndFetchProjectionWeights()
1534 const CudnnHandle& cudnn, int input_size, cudnnDataType_t data_type, in Create()
[all …]
|