Home
last modified time | relevance | path

Searched refs:cudnn_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc332 cudnn_.reset(new CudnnAccess(cudnn_handle)); in Init()
1770 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoRnnForwardImpl()
1917 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoRnnBackwardImpl()
2038 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoCtcLossImpl()
2079 auto cudnn = cudnn_->GetHandle(parent_, /*stream=*/nullptr); in createRnnDescriptor()
3003 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoPrepareForConvolution()
3060 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoConvolve()
3209 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoFusedConvolveImpl()
3517 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoBatchNormalizationForwardImpl()
3699 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoBatchNormalizationBackwardImpl()
[all …]
DBUILD373 textual_hdrs = glob(["cudnn_*.inc"]),
Dcuda_dnn.h585 std::unique_ptr<class CudnnAccess> cudnn_; variable