Home
last modified time | relevance | path

Searched refs:SetDefaultFactory (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.h83 bool SetDefaultFactory(Platform::Id platform_id, PluginKind plugin_kind,
Dplugin_registry.cc107 bool PluginRegistry::SetDefaultFactory(Platform::Id platform_id, in SetDefaultFactory() function in stream_executor::PluginRegistry
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_rng.cc259 PluginRegistry::Instance()->SetDefaultFactory( in initialize_curand()
Dcuda_fft.cc585 PluginRegistry::Instance()->SetDefaultFactory( in initialize_cufft()
Dcuda_dnn.cc4678 PluginRegistry::Instance()->SetDefaultFactory( in initialize_cudnn()
Dcuda_blas.cc3785 PluginRegistry::Instance()->SetDefaultFactory( in initialize_cublas()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_rng.cc316 PluginRegistry::Instance()->SetDefaultFactory( in initialize_rocrand()
Drocm_fft.cc615 PluginRegistry::Instance()->SetDefaultFactory( in initialize_rocfft()
Drocm_blas.cc2719 PluginRegistry::Instance()->SetDefaultFactory( in initialize_rocblas()
Drocm_dnn.cc5110 PluginRegistry::Instance()->SetDefaultFactory( in initialize_miopen()