Home
last modified time | relevance | path

Searched refs:kDnn (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.cc34 case PluginKind::kDnn: in PluginKindString()
130 case PluginKind::kDnn: in SetDefaultFactory()
154 case PluginKind::kDnn: in HasFactory()
Dplugin.h40 kDnn, enumerator
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc5080 rocm::kROCmPlatformId, PluginKind::kDnn, gpu::kMIOpenPlugin); in initialize_miopen()
5111 rocm::kROCmPlatformId, PluginKind::kDnn, gpu::kMIOpenPlugin); in initialize_miopen()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4679 cuda::kCudaPlatformId, PluginKind::kDnn, gpu::kCuDnnPlugin); in initialize_cudnn()