Searched defs:cudnn (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 993 const CudnnHandle& cudnn, float dropout, uint64 seed, in Create()
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()
1659 const CudnnHandle& cudnn, const CudnnRnnDescriptor& rnn_desc, in CheckRNNParameterSize()
1675 Stream* stream, const CudnnHandle& cudnn, in CreateRnnWorkspace()
1694 Stream* stream, const CudnnHandle& cudnn, const cudnnBatchNormMode_t& mode, in CreateBatchNormForwardWorkspace()
1718 Stream* stream, const CudnnHandle& cudnn, const cudnnBatchNormMode_t& mode, in CreateBatchNormBackwardWorkspace()
1770 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
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | cudnn_recurrent_test.py | 464 def gru(cudnn=False, **kwargs): argument
|