Searched defs:dnn (Results 1 – 10 of 10) sorted by relevance
108 std::map<PluginId, DnnFactory> dnn; member117 PluginId blas, dnn, fft, rng; member
38 PluginConfig& PluginConfig::SetDnn(PluginId dnn) { in SetDnn()
77 PluginId dnn() const { return dnn_; } in dnn() function
359 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local387 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local418 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local447 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local481 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm() local519 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm() local558 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm() local596 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm() local634 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenFusedConvolveWithAlgorithm() local667 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveWithAlgorithm() local[all …]
23 namespace dnn { namespace
94 if (auto* dnn = stream_executor->AsDnn()) { in HeuristicLayoutAssignment() local
158 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local189 auto* dnn = se->AsDnn(); in PrintPlatformInfo() local
101 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local
4788 gpu::MIOpenSupport* dnn = new gpu::MIOpenSupport(rocm_executor); in initialize_miopen() local
4673 gpu::CudnnSupport* dnn = new gpu::CudnnSupport(cuda_executor); in initialize_cudnn() local