Searched refs:AsDnn (Results 1 – 8 of 8) sorted by relevance
267 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveAlgorithms()286 dnn::DnnSupport *dnn_support = AsDnn(); in GetMIOpenConvolveAlgorithms()298 dnn::DnnSupport *dnn_support = AsDnn(); in GetRnnAlgorithms()308 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardDataAlgorithms()321 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardFilterAlgorithms()372 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnDescriptor()387 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnSequenceTensorDescriptor()401 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnSequenceTensorDescriptor()415 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnStateTensorDescriptor()424 dnn::DnnSupport *StreamExecutor::AsDnn() { in AsDnn() function in stream_executor::StreamExecutor
358 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward()382 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward()409 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward()434 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward()466 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm()497 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm()529 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm()560 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm()591 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm()635 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveQuantized()[all …]
494 dnn::DnnSupport *AsDnn();
338 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveWithAlgorithm()449 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveBackwardDataWithAlgorithm()481 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveBackwardFilterWithAlgorithm()
165 auto* dnn_support = stream_exec->AsDnn(); in ShouldIncludeWinogradNonfusedAlgo()
101 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion()
100 if (auto* dnn = stream_executor->AsDnn()) { in HeuristicLayoutAssignment()
166 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion()197 auto* dnn = se->AsDnn(); in PrintPlatformInfo()