Searched defs:cudnn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1167 const CudnnHandle& cudnn, float dropout, uint64_t seed, in Create()
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()
1834 const CudnnHandle& cudnn, const CudnnRnnDescriptor& rnn_desc, in CheckRNNParameterSize()
1856 Stream* stream, const CudnnHandle& cudnn, in CreateRnnWorkspace()
1875 Stream* stream, const CudnnHandle& cudnn, const cudnnBatchNormMode_t& mode, in CreateBatchNormForwardWorkspace()
1899 Stream* stream, const CudnnHandle& cudnn, const cudnnBatchNormMode_t& mode, in CreateBatchNormBackwardWorkspace()
1954 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoRnnForwardImpl() local
[all …]
|
| /external/tensorflow/tensorflow/tools/tensorflow_builder/compat_checker/ |
| D | sample_config.ini | 5 cudnn = [range(7.0.0, 8.0.0)] key
|