Searched refs:cudnn_rnn_desc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2144 const CudnnRnnDescriptor& cudnn_rnn_desc = in DoRnnForward() local
2160 stream, cudnn_rnn_desc, cudnn_input_desc, input_data, in DoRnnForward()
2185 const CudnnRnnDescriptor& cudnn_rnn_desc = in DoRnnForward() local
2201 stream, cudnn_rnn_desc, cudnn_input_desc, input_data, in DoRnnForward()
2227 const CudnnRnnDescriptor& cudnn_rnn_desc = in DoRnnForward() local
2243 stream, cudnn_rnn_desc, cudnn_input_desc, input_data, in DoRnnForward()
2276 const CudnnRnnDescriptor& cudnn_rnn_desc = in DoRnnBackward() local
2292 stream, cudnn_rnn_desc, cudnn_input_desc, input_data, in DoRnnBackward()
2327 const CudnnRnnDescriptor& cudnn_rnn_desc = in DoRnnBackward() local
2343 stream, cudnn_rnn_desc, cudnn_input_desc, input_data, in DoRnnBackward()
[all …]
|