Home
last modified time | relevance | path

Searched refs:RngFactory (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.h56 typedef rng::RngSupport* (*RngFactory)(internal::StreamExecutorInterface*); typedef
111 std::map<PluginId, RngFactory> rng;
Dplugin_registry.cc245 EMIT_PLUGIN_SPECIALIZATIONS(RngFactory, rng, "RNG");
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc248 ->GetFactory<PluginRegistry::RngFactory>(kHostPlatformId, in SupportsRng()
255 port::StatusOr<PluginRegistry::RngFactory> status = in CreateRng()
256 registry->GetFactory<PluginRegistry::RngFactory>(kHostPlatformId, in CreateRng()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_rng.cc233 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::RngFactory>( in initialize_curand()
Dcuda_gpu_executor.cc818 port::StatusOr<PluginRegistry::RngFactory> status = in CreateRng()
819 registry->GetFactory<PluginRegistry::RngFactory>(cuda::kCudaPlatformId, in CreateRng()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_rng.cc291 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::RngFactory>( in initialize_rocrand()
Drocm_gpu_executor.cc699 port::StatusOr<PluginRegistry::RngFactory> status = in CreateRng()
700 registry->GetFactory<PluginRegistry::RngFactory>(rocm::kROCmPlatformId, in CreateRng()