Searched refs:CudnnRnnDescriptor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1002 class CudnnRnnDescriptor : public dnn::RnnDescriptor { class
1003 CudnnRnnDescriptor(const CudnnHandle& cudnn, gpu::RnnDescriptor rnn_desc, in CudnnRnnDescriptor() function in stream_executor::gpu::CudnnRnnDescriptor
1030 CudnnRnnDescriptor(CudnnRnnDescriptor&& other) = default;
1032 static port::StatusOr<CudnnRnnDescriptor> Create( in Create()
1069 return port::StatusOr<CudnnRnnDescriptor>(rnn_plan_wrapper.status()); in Create()
1100 return CudnnRnnDescriptor(cudnn, std::move(rnn_desc), std::move(rnn_plan), in Create()
1151 SE_DISALLOW_COPY_AND_ASSIGN(CudnnRnnDescriptor);
1412 const CudnnRnnDescriptor& rnn_desc, in ExtractAndCheckRnnForward()
1470 const CudnnHandle& cudnn, const CudnnRnnDescriptor& rnn_desc, in CheckRNNParameterSize()
1487 const CudnnRnnDescriptor& rnn_desc, in CreateRnnWorkspace()
[all …]
|
| D | cuda_dnn.h | 34 class CudnnRnnDescriptor; variable
607 Stream* stream, const CudnnRnnDescriptor& rnn_desc,
626 Stream* stream, const CudnnRnnDescriptor& rnn_desc,
|