Searched refs:CudnnSupport (Results 1 – 2 of 2) sorted by relevance
307 CudnnSupport::CudnnSupport(GpuExecutor* parent) : parent_(parent) {} in CudnnSupport() function in stream_executor::gpu::CudnnSupport309 port::Status CudnnSupport::Init() { in Init()358 CudnnSupport::GetVersion() { in GetVersion()1746 port::Status CudnnSupport::DoRnnForwardImpl( in DoRnnForwardImpl()1886 port::Status CudnnSupport::DoRnnBackwardImpl( in DoRnnBackwardImpl()2030 port::Status CudnnSupport::DoCtcLossImpl( in DoCtcLossImpl()2070 CudnnSupport::createRnnDescriptor( in createRnnDescriptor()2093 CudnnSupport::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()2105 CudnnSupport::createRnnSequenceTensorDescriptor( in createRnnSequenceTensorDescriptor()2118 CudnnSupport::createRnnStateTensorDescriptor(int num_layer, int batch_size, in createRnnStateTensorDescriptor()[all …]
43 class CudnnSupport : public dnn::DnnSupport {45 explicit CudnnSupport(GpuExecutor* parent);716 SE_DISALLOW_COPY_AND_ASSIGN(CudnnSupport);