Home
last modified time | relevance | path

Searched refs:host_to_device (Results 1 – 4 of 4) 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.cc268 group->host_to_device = GetStream(executor, priority); in GetOrCreate()
269 group->host_to_device->Init(); in GetOrCreate()
271 << "] = " << group->host_to_device; in GetOrCreate()
320 if (stream.host_to_device) { in TestOnlyReset()
321 delete stream.host_to_device; in TestOnlyReset()
322 stream.host_to_device = nullptr; in TestOnlyReset()
441 stream_->host_to_device, stream_->device_to_host, in Init()
Dgpu_device.h136 se::Stream* host_to_device = nullptr; member
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc160 se::Stream* h2d_stream() { return gpu_->stream_->host_to_device; } in h2d_stream()