Searched defs:dnn (Results 1 – 11 of 11) sorted by relevance
109 std::map<PluginId, DnnFactory> dnn; member118 PluginId blas, dnn, fft, rng; member
38 PluginConfig& PluginConfig::SetDnn(PluginId dnn) { in SetDnn()
77 PluginId dnn() const { return dnn_; } in dnn() function
354 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local378 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local404 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local429 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local461 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm() local499 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm() local538 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm() local576 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm() local609 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveWithAlgorithm() local651 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveWithAlgorithm() local[all …]
23 namespace dnn { namespace
33 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local
78 if (auto* dnn = stream_executor->AsDnn()) { in HeuristicLayoutAssignment() local
102 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local
275 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local
4131 gpu::CudnnSupport* dnn = new gpu::CudnnSupport(cuda_executor); in initialize_cudnn() local
4500 gpu::MIOpenSupport* dnn = new gpu::MIOpenSupport(rocm_executor); in initialize_miopen() local