Searched defs:dnn (Results 1 – 13 of 13) sorted by relevance
108 std::map<PluginId, DnnFactory> dnn; member117 PluginId blas, dnn, fft, rng; member
343 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local369 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local397 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local424 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local471 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveQuantized() local498 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveQuantized() local525 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenSeparableConvolve() local544 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenMatMul() local563 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenMatMulQuantized() local583 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenMatMulQuantized() local[all …]
38 PluginConfig& PluginConfig::SetDnn(PluginId dnn) { in SetDnn()
77 PluginId dnn() const { return dnn_; } in dnn() function
349 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveWithAlgorithm() local380 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm() local475 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { variable494 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { variable
25 namespace dnn { namespace
778 dnn::DnnSupport* dnn; in DeallocateStream() local
221 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local251 auto* dnn = se->AsDnn(); in PrintPlatformInfo() local
124 if (auto* dnn = stream_executor->AsDnn()) { in HeuristicLayoutAssignment() local
255 if (se::dnn::DnnSupport* dnn = stream_exec_->AsDnn()) { in IsConvBF16Supported() local
113 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local
5035 gpu::MIOpenSupport* dnn = new gpu::MIOpenSupport(rocm_executor); in initialize_miopen() local
6575 gpu::CudnnSupport* dnn = new gpu::CudnnSupport(cuda_executor); in initialize_cudnn() local