Searched refs:CudnnAccess (Results 1 – 2 of 2) sorted by relevance
185 class CudnnAccess { class188 explicit CudnnAccess(cudnnHandle_t handle) : handle_(handle) {} in CudnnAccess() function in stream_executor::gpu::CudnnAccess190 ~CudnnAccess() { in ~CudnnAccess()424 cudnn_.reset(new CudnnAccess(cudnn_handle)); in Init()4015 GpuExecutor* parent, Stream* stream, CudnnAccess* cudnn, in Create()4215 CudnnLegacyConvRunner(GpuExecutor* parent, CudnnAccess* cudnn, in CudnnLegacyConvRunner()4242 CudnnAccess* cudnn_;4525 GpuExecutor* parent, CudnnAccess* cudnn, in Create()4535 CudnnExecutionPlanRunner(GpuExecutor* parent, CudnnAccess* cudnn, in CudnnExecutionPlanRunner()4546 CudnnAccess* cudnn_;[all …]
534 std::unique_ptr<class CudnnAccess> cudnn_;