MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / CreateRng

Method CreateRng

tensorflow/stream_executor/host/host_gpu_executor.cc:332–344  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

330}
331
332rng::RngSupport *HostExecutor::CreateRng() {
333 PluginRegistry *registry = PluginRegistry::Instance();
334 port::StatusOr<PluginRegistry::RngFactory> status =
335 registry->GetFactory<PluginRegistry::RngFactory>(kHostPlatformId,
336 plugin_config_.rng());
337 if (!status.ok()) {
338 LOG(ERROR) << "Unable to retrieve RNG factory: "
339 << status.status().error_message();
340 return nullptr;
341 }
342
343 return status.ValueOrDie()(this);
344}
345
346std::unique_ptr<internal::StreamInterface>
347HostExecutor::GetStreamImplementation() {

Callers 1

AsRngMethod · 0.45

Calls 3

rngMethod · 0.45
okMethod · 0.45
statusMethod · 0.45

Tested by

no test coverage detected