Searched defs:dnn (Results 1 – 12 of 12) sorted by relevance
108 std::map<PluginId, DnnFactory> dnn; member117 PluginId blas, dnn, fft, rng; member
343 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveWithAlgorithm() local371 dnn::DnnSupport *dnn = parent_->AsDnn(); in ConvolveWithExecutionPlan() local400 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm() local429 dnn::DnnSupport *dnn = parent_->AsDnn(); in FusedConvolveWithExecutionPlan() local466 dnn::DnnSupport *dnn = parent_->AsDnn(); in ConvolveBackwardDataWithExecutionPlan() local495 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveBackwardDataWithAlgorithm() local527 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveBackwardFilterWithAlgorithm() local558 dnn::DnnSupport *dnn = parent_->AsDnn(); in ConvolveBackwardFilterWithExecutionPlan() local
38 PluginConfig& PluginConfig::SetDnn(PluginId dnn) { in SetDnn()
77 PluginId dnn() const { return dnn_; } in dnn() function
357 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local381 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local408 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local433 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local478 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveQuantized() local505 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveQuantized() local532 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenSeparableConvolve() local551 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenMatMul() local570 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenMatMulQuantized() local590 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenMatMulQuantized() local[all …]
23 namespace dnn { namespace
160 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local190 auto* dnn = se->AsDnn(); in PrintPlatformInfo() local
119 if (auto* dnn = stream_executor->AsDnn()) { in HeuristicLayoutAssignment() local
209 if (se::dnn::DnnSupport* dnn = stream_exec_->AsDnn()) { in IsConvBF16Supported() local
101 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local
4986 gpu::MIOpenSupport* dnn = new gpu::MIOpenSupport(rocm_executor); in initialize_miopen() local
5874 gpu::CudnnSupport* dnn = new gpu::CudnnSupport(cuda_executor); in initialize_cudnn() local