Searched refs:CudnnRnnStateTensorDescriptor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1551 class CudnnRnnStateTensorDescriptor : public dnn::RnnStateTensorDescriptor { class
1553 CudnnRnnStateTensorDescriptor(GpuExecutor* parent, int num_layers, in CudnnRnnStateTensorDescriptor() function in stream_executor::gpu::CudnnRnnStateTensorDescriptor
1581 SE_DISALLOW_COPY_AND_ASSIGN(CudnnRnnStateTensorDescriptor);
1601 const CudnnRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward()
1603 const CudnnRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward()
1607 const CudnnRnnStateTensorDescriptor& output_h_desc, in ExtractAndCheckRnnForward()
1609 const CudnnRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward()
1750 const CudnnRnnStateTensorDescriptor& input_h_desc, in DoRnnForwardImpl()
1752 const CudnnRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl()
1756 const CudnnRnnStateTensorDescriptor& output_h_desc, in DoRnnForwardImpl()
[all …]
|
| D | cuda_dnn.h | 35 class CudnnRnnStateTensorDescriptor; variable
644 const CudnnRnnStateTensorDescriptor& input_h_desc,
646 const CudnnRnnStateTensorDescriptor& input_c_desc,
650 const CudnnRnnStateTensorDescriptor& output_h_desc,
652 const CudnnRnnStateTensorDescriptor& output_c_desc,
663 const CudnnRnnStateTensorDescriptor& input_h_desc,
665 const CudnnRnnStateTensorDescriptor& input_c_desc,
669 const CudnnRnnStateTensorDescriptor& output_h_desc,
671 const CudnnRnnStateTensorDescriptor& output_c_desc,
685 Stream* stream, const CudnnRnnStateTensorDescriptor& probs_desc,
[all …]
|