Searched refs:AsDnn (Results 1 – 8 of 8) sorted by relevance
326 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveAlgorithms()338 dnn::DnnSupport *dnn_support = AsDnn(); in GetRnnAlgorithms()348 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardDataAlgorithms()361 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardFilterAlgorithms()387 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnDescriptor()402 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnSequenceTensorDescriptor()416 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnSequenceTensorDescriptor()430 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnStateTensorDescriptor()439 dnn::DnnSupport *StreamExecutor::AsDnn() { in AsDnn() function in stream_executor::StreamExecutor
354 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward()378 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward()404 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward()429 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward()461 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm()499 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm()538 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm()576 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm()609 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveWithAlgorithm()651 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveWithAlgorithm()[all …]
472 dnn::DnnSupport *AsDnn();
33 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion()
145 auto* dnn_support = stream_exec->AsDnn(); in ShouldIncludeWinogradNonfusedAlgo()
78 if (auto* dnn = stream_executor->AsDnn()) { in HeuristicLayoutAssignment()
102 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion()
275 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion()