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

Function AsHostStream

tensorflow/stream_executor/host/host_gpu_executor.cc:36–39  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

34namespace host {
35
36HostStream *AsHostStream(Stream *stream) {
37 DCHECK(stream != nullptr);
38 return dynamic_cast<HostStream *>(stream->implementation());
39}
40
41HostExecutor::HostExecutor(const PluginConfig &plugin_config)
42 : plugin_config_(plugin_config) {}

Callers 10

MemcpyMethod · 0.85
MemcpyDeviceToDeviceMethod · 0.85
MemZeroMethod · 0.85
MemsetMethod · 0.85
Memset32Method · 0.85
HostCallbackMethod · 0.85
RecordEventMethod · 0.85
WaitForEventMethod · 0.85
BlockHostUntilDoneMethod · 0.85

Calls 1

implementationMethod · 0.45

Tested by

no test coverage detected