Searched defs:dnn (Results 1 – 11 of 11) 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
338 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveWithAlgorithm() local449 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveBackwardDataWithAlgorithm() local481 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveBackwardFilterWithAlgorithm() local
358 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local382 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local409 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local434 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local466 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm() local497 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm() local529 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm() local560 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm() local591 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm() local635 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveQuantized() local[all …]
23 namespace dnn { namespace
100 if (auto* dnn = stream_executor->AsDnn()) { in HeuristicLayoutAssignment() local
166 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local197 auto* dnn = se->AsDnn(); in PrintPlatformInfo() local
101 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local
4664 gpu::CudnnSupport* dnn = new gpu::CudnnSupport(cuda_executor); in initialize_cudnn() local
5096 gpu::MIOpenSupport* dnn = new gpu::MIOpenSupport(rocm_executor); in initialize_miopen() local