Searched refs:CudnnRnnDescriptor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1057 class CudnnRnnDescriptor : public dnn::RnnDescriptor { class
1058 CudnnRnnDescriptor(const CudnnHandle& cudnn, gpu::RnnDescriptor rnn_desc, in CudnnRnnDescriptor() function in stream_executor::gpu::CudnnRnnDescriptor
1086 CudnnRnnDescriptor(CudnnRnnDescriptor&& other) = default;
1088 static port::StatusOr<CudnnRnnDescriptor> Create( in Create()
1180 return port::StatusOr<CudnnRnnDescriptor>(rnn_plan_wrapper.status()); in Create()
1194 return CudnnRnnDescriptor(cudnn, std::move(rnn_desc), std::move(rnn_plan), in Create()
1249 SE_DISALLOW_COPY_AND_ASSIGN(CudnnRnnDescriptor);
1598 const CudnnRnnDescriptor& rnn_desc, in ExtractAndCheckRnnForward()
1659 const CudnnHandle& cudnn, const CudnnRnnDescriptor& rnn_desc, in CheckRNNParameterSize()
1676 const CudnnRnnDescriptor& rnn_desc, in CreateRnnWorkspace()
[all …]
|
| D | cuda_dnn.h | 33 class CudnnRnnDescriptor; variable
641 Stream* stream, const CudnnRnnDescriptor& rnn_desc,
660 Stream* stream, const CudnnRnnDescriptor& rnn_desc,
|