Searched refs:device_to_host (Results 1 – 6 of 6) sorted by relevance
| /external/mesa3d/src/gallium/drivers/r600/ |
| D | compute_memory_pool.c | 49 struct pipe_context *pipe, int device_to_host);
65 struct pipe_context * pipe, int device_to_host,
195 struct pipe_context * pipe, int device_to_host) in compute_memory_shadow() argument
200 device_to_host); in compute_memory_shadow()
205 compute_memory_transfer(pool, pipe, device_to_host, &chunk, in compute_memory_shadow()
597 int device_to_host, in compute_memory_transfer() argument
613 "offset_in_chunk = %d, size = %d\n", device_to_host, in compute_memory_transfer()
616 if (device_to_host) { in compute_memory_transfer()
|
| /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/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 273 group->device_to_host = GetStream(executor, priority); in GetOrCreate()
274 group->device_to_host->Init(); in GetOrCreate()
276 << "] = " << group->device_to_host; in GetOrCreate()
324 if (stream.device_to_host) { in TestOnlyReset()
325 delete stream.device_to_host; in TestOnlyReset()
326 stream.device_to_host = nullptr; in TestOnlyReset()
441 stream_->host_to_device, stream_->device_to_host, in Init()
|
| D | gpu_device.h | 137 se::Stream* device_to_host = nullptr; member
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 225 Type type, bool device_to_host) { in SetFrontendAttributes() argument
229 device_to_host ? llvm::formatv("{0}_dtoh_{1}", key, index).str() in SetFrontendAttributes()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 161 se::Stream* d2h_stream() { return gpu_->stream_->device_to_host; } in d2h_stream()
|