Searched refs:device_to_host (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | host_compute_metadata.proto | 33 // Metadata about each device_to_host transfer
34 repeated HostTransferMetadata device_to_host = 1; field
|
| /external/mesa3d/src/gallium/drivers/r600/ |
| D | compute_memory_pool.h | 99 struct pipe_context *pipe, int device_to_host);
126 struct pipe_context * pipe, int device_to_host,
|
| D | compute_memory_pool.c | 255 struct pipe_context * pipe, int device_to_host) in compute_memory_shadow() argument
260 device_to_host); in compute_memory_shadow()
265 compute_memory_transfer(pool, pipe, device_to_host, &chunk, in compute_memory_shadow()
658 int device_to_host, in compute_memory_transfer() argument
674 "offset_in_chunk = %d, size = %d\n", device_to_host, in compute_memory_transfer()
677 if (device_to_host) { in compute_memory_transfer()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.h | 141 se::Stream* device_to_host = nullptr; member
|
| D | gpu_device.cc | 255 group->device_to_host = new se::Stream(executor); in GetOrCreate()
256 group->device_to_host->Init(); in GetOrCreate()
258 << "] = " << group->device_to_host; in GetOrCreate()
392 streams_.back()->device_to_host, streams_.back()->device_to_device)); in Init()
|