Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.h55 typedef rng::RngSupport* (*RngFactory)(internal::StreamExecutorInterface*); typedef
110 std::map<PluginId, RngFactory> rng;
179 DECLARE_PLUGIN_SPECIALIZATIONS(RngFactory);
Dplugin_registry.cc249 EMIT_PLUGIN_SPECIALIZATIONS(RngFactory, rng, "RNG");
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc340 ->GetFactory<PluginRegistry::RngFactory>(kHostPlatformId, in SupportsRng()
347 port::StatusOr<PluginRegistry::RngFactory> status = in CreateRng()
348 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.cc790 port::StatusOr<PluginRegistry::RngFactory> status = in CreateRng()
791 registry->GetFactory<PluginRegistry::RngFactory>(cuda::kCudaPlatformId, in CreateRng()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_rng.cc290 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::RngFactory>( in initialize_rocrand()
Drocm_gpu_executor.cc698 port::StatusOr<PluginRegistry::RngFactory> status = in CreateRng()
699 registry->GetFactory<PluginRegistry::RngFactory>(rocm::kROCmPlatformId, in CreateRng()