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.cc324 cudnn_.reset(new CudnnAccess(cudnn_handle)); in Init()
1530 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoRnnForwardImpl()
1690 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoRnnBackwardImpl()
1824 auto cudnn = cudnn_->GetHandle(parent_, /*stream=*/nullptr); in createRnnDescriptor()
2648 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoPrepareForConvolution()
2700 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoConvolve()
2956 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoFusedConvolveImpl()
3244 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoBatchNormalizationForwardImpl()
3331 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoBatchNormalizationBackwardImpl()
3460 auto cudnn = cudnn_->GetHandle(parent_, stream); in DoTransformTensor()
[all …]
DBUILD261 textual_hdrs = glob(["cudnn_*.inc"]),
Dcuda_dnn.h554 std::unique_ptr<class CudnnAccess> cudnn_; variable