Home
last modified time | relevance | path

Searched refs:DnnFactory (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.h54 typedef dnn::DnnSupport* (*DnnFactory)(internal::StreamExecutorInterface*); typedef
109 std::map<PluginId, DnnFactory> dnn;
Dplugin_registry.cc243 EMIT_PLUGIN_SPECIALIZATIONS(DnnFactory, dnn, "DNN");
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc671 port::StatusOr<PluginRegistry::DnnFactory> status = in CreateDnn()
672 registry->GetFactory<PluginRegistry::DnnFactory>(rocm::kROCmPlatformId, in CreateDnn()
Drocm_dnn.cc4488 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::DnnFactory>( in initialize_miopen()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc790 port::StatusOr<PluginRegistry::DnnFactory> status = in CreateDnn()
791 registry->GetFactory<PluginRegistry::DnnFactory>(cuda::kCudaPlatformId, in CreateDnn()
Dcuda_dnn.cc4120 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::DnnFactory>( in initialize_cudnn()