Home
last modified time | relevance | path

Searched refs:host_to_device (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dhost_compute_metadata.proto36 // Metadata about each host_to_device transfer
37 repeated HostTransferMetadata host_to_device = 2; field
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.h140 se::Stream* host_to_device = nullptr; member
Dgpu_device.cc250 group->host_to_device = new se::Stream(executor); in GetOrCreate()
251 group->host_to_device->Init(); in GetOrCreate()
253 << "] = " << group->host_to_device; in GetOrCreate()
391 i, streams_.back()->compute, streams_.back()->host_to_device, in Init()