Searched refs:cudnn_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 324 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 …]
|
| D | BUILD | 261 textual_hdrs = glob(["cudnn_*.inc"]),
|
| D | cuda_dnn.h | 554 std::unique_ptr<class CudnnAccess> cudnn_; variable
|