Searched refs:cudnn_input_desc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2490 const CudnnRnnSequenceTensorDescriptor& cudnn_input_desc = in DoRnnForward() local
2504 stream, cudnn_rnn_desc, cudnn_input_desc, input_data, in DoRnnForward()
2533 const CudnnRnnSequenceTensorDescriptor& cudnn_input_desc = in DoRnnForward() local
2547 stream, cudnn_rnn_desc, cudnn_input_desc, input_data, in DoRnnForward()
2577 const CudnnRnnSequenceTensorDescriptor& cudnn_input_desc = in DoRnnForward() local
2591 stream, cudnn_rnn_desc, cudnn_input_desc, input_data, in DoRnnForward()
2628 const CudnnRnnSequenceTensorDescriptor& cudnn_input_desc = in DoRnnBackward() local
2642 stream, cudnn_rnn_desc, cudnn_input_desc, input_data, in DoRnnBackward()
2680 const CudnnRnnSequenceTensorDescriptor& cudnn_input_desc = in DoRnnBackward() local
2694 stream, cudnn_rnn_desc, cudnn_input_desc, input_data, in DoRnnBackward()
[all …]
|