Searched refs:sync_dst_compute (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 59 bool sync_dst_compute) { in CopyHostToDevice() argument
70 cpu_allocator, edge_name, sync_dst_compute, in CopyHostToDevice()
76 to, recv_dev_context, wrapped_done, sync_dst_compute); in CopyHostToDevice()
91 sync_dst_compute); in CopyHostToDevice()
118 sync_dst_compute); in CopyHostToDevice()
208 bool sync_dst_compute) { in ViaDMA() argument
265 sync_dst_compute](Status status) { in ViaDMA()
272 std::move(delete_and_done), sync_dst_compute); in ViaDMA()
293 sync_dst_compute); in ViaDMA()
|
| D | rendezvous_mgr.cc | 98 bool sync_dst_compute = true; in SameWorkerRecvDone() local
114 sync_dst_compute = false; in SameWorkerRecvDone()
129 out, 0 /*dev_to_dev_stream_index*/, std::move(done), sync_dst_compute); in SameWorkerRecvDone()
|
| D | copy_tensor.h | 49 bool sync_dst_compute = true);
|
| D | gpu_device_context.h | 69 bool sync_dst_compute) const override;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 30 bool sync_dst_compute) const { in CopyCPUTensorToDevice()
32 sync_dst_compute); in CopyCPUTensorToDevice()
|
| D | gpu_util.h | 91 StatusCallback done, bool sync_dst_compute);
|
| D | gpu_util.cc | 314 StatusCallback done, bool sync_dst_compute) { in CopyCPUTensorToGPU() argument
333 if (sync_dst_compute) { in CopyCPUTensorToGPU()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_context.cc | 29 StatusCallback done, bool sync_dst_compute) const { in CopyCPUTensorToDevice()
31 cpu_tensor, this, device, device_tensor, done, sync_dst_compute); in CopyCPUTensorToDevice()
|
| D | pluggable_device_util.h | 52 bool sync_dst_compute);
|
| D | pluggable_device_util.cc | 224 bool sync_dst_compute) { in CopyCPUTensorToPluggableDevice() argument
244 if (sync_dst_compute) { in CopyCPUTensorToPluggableDevice()
|
| D | pluggable_device_context.h | 54 bool sync_dst_compute) const override;
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 29 bool sync_dst_compute) const override;
42 bool sync_dst_compute) const { in CopyCPUTensorToDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 284 bool sync_dst_compute = (safe_alloc_frontier == 0); in SameWorkerRecvDone() local
289 if (!sync_dst_compute) { in SameWorkerRecvDone()
303 out, 0 /*dev_to_dev_stream_index*/, std::move(done), sync_dst_compute); in SameWorkerRecvDone()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 67 bool sync_dst_compute) const override;
|
| D | xla_device_context.cc | 118 bool sync_dst_compute) const { in CopyCPUTensorToDevice()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 81 bool sync_dst_compute = true) const {
|