Searched defs:dnn (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | plugin_registry.h | 108 std::map<PluginId, DnnFactory> dnn; member
117 PluginId blas, dnn, fft, rng; member
|
| D | stream.cc | 343 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local
369 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local
397 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward() local
424 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward() local
471 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveQuantized() local
498 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveQuantized() local
525 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenSeparableConvolve() local
544 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenMatMul() local
563 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenMatMulQuantized() local
583 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenMatMulQuantized() local
[all …]
|
| D | plugin.cc | 38 PluginConfig& PluginConfig::SetDnn(PluginId dnn) { in SetDnn()
|
| D | plugin.h | 77 PluginId dnn() const { return dnn_; } in dnn() function
|
| D | stream.h | 349 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveWithAlgorithm() local
380 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm() local
475 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { variable
494 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { variable
|
| D | dnn.cc | 25 namespace dnn { namespace
|
| D | stream_executor_pimpl.cc | 778 dnn::DnnSupport* dnn; in DeallocateStream() local
|
| /external/pytorch/test/optim/ |
| D | test_swa_utils.py | 81 def _run_averaged_steps(self, dnn, swa_device, ema): argument
225 def _test_update_bn(self, dnn, dl_x, dl_xy, cuda): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 221 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local
251 auto* dnn = se->AsDnn(); in PrintPlatformInfo() local
|
| D | gpu_layout_assignment.cc | 124 if (auto* dnn = stream_executor->AsDnn()) { in HeuristicLayoutAssignment() local
|
| D | gpu_compiler.cc | 255 if (se::dnn::DnnSupport* dnn = stream_exec_->AsDnn()) { in IsConvBF16Supported() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 113 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local
|
| /external/libopus/ |
| D | meson.build | 642 subdir('dnn') subdir
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 5035 gpu::MIOpenSupport* dnn = new gpu::MIOpenSupport(rocm_executor); in initialize_miopen() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 6575 gpu::CudnnSupport* dnn = new gpu::CudnnSupport(cuda_executor); in initialize_cudnn() local
|