Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.h53 typedef dnn::DnnSupport* (*DnnFactory)(internal::StreamExecutorInterface*); typedef
108 std::map<PluginId, DnnFactory> dnn;
177 DECLARE_PLUGIN_SPECIALIZATIONS(DnnFactory);
Dplugin_registry.cc247 EMIT_PLUGIN_SPECIALIZATIONS(DnnFactory, dnn, "DNN");
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc670 port::StatusOr<PluginRegistry::DnnFactory> status = in CreateDnn()
671 registry->GetFactory<PluginRegistry::DnnFactory>(rocm::kROCmPlatformId, in CreateDnn()
Drocm_dnn.cc5084 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::DnnFactory>( in initialize_miopen()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc762 port::StatusOr<PluginRegistry::DnnFactory> status = in CreateDnn()
763 registry->GetFactory<PluginRegistry::DnnFactory>(cuda::kCudaPlatformId, in CreateDnn()
Dcuda_dnn.cc4653 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::DnnFactory>( in initialize_cudnn()