Searched refs:CudnnSupport (Results 1 – 2 of 2) sorted by relevance
298 CudnnSupport::CudnnSupport(GpuExecutor* parent) : parent_(parent) {} in CudnnSupport() function in stream_executor::gpu::CudnnSupport300 port::Status CudnnSupport::Init() { in Init()348 CudnnSupport::GetVersion() { in GetVersion()1506 port::Status CudnnSupport::DoRnnForwardImpl( in DoRnnForwardImpl()1659 port::Status CudnnSupport::DoRnnBackwardImpl( in DoRnnBackwardImpl()1816 CudnnSupport::createRnnDescriptor( in createRnnDescriptor()1838 CudnnSupport::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor()1850 CudnnSupport::createRnnSequenceTensorDescriptor( in createRnnSequenceTensorDescriptor()1863 CudnnSupport::createRnnStateTensorDescriptor(int num_layer, int batch_size, in createRnnStateTensorDescriptor()1871 bool CudnnSupport::DoRnnForward( in DoRnnForward()[all …]
43 class CudnnSupport : public dnn::DnnSupport {45 explicit CudnnSupport(GpuExecutor* parent);663 SE_DISALLOW_COPY_AND_ASSIGN(CudnnSupport);