Searched refs:sync_dst_compute (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 55 bool sync_dst_compute) { in CopyHostToDevice() argument
67 edge_name, sync_dst_compute, wrapped_done = std::move(wrapped_done)]( in CopyHostToDevice()
73 sync_dst_compute); in CopyHostToDevice()
88 &from, dst, to, wrapped_done, sync_dst_compute); in CopyHostToDevice()
115 sync_dst_compute); in CopyHostToDevice()
206 bool sync_dst_compute) { in ViaDMA() argument
258 sync_dst_compute](Status status) { in ViaDMA()
265 std::move(delete_and_done), sync_dst_compute); in ViaDMA()
286 sync_dst_compute); in ViaDMA()
|
| D | rendezvous_mgr.cc | 83 bool sync_dst_compute = true; in SameWorkerRecvDone() local
97 sync_dst_compute = false; in SameWorkerRecvDone()
112 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 | 64 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 | 302 StatusCallback done, bool sync_dst_compute) { in CopyCPUTensorToGPU() argument
321 if (sync_dst_compute) { in CopyCPUTensorToGPU()
|
| /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 | 270 bool sync_dst_compute = (safe_alloc_frontier == 0); in SameWorkerRecvDone() local
275 if (!sync_dst_compute) { in SameWorkerRecvDone()
289 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 | 65 bool sync_dst_compute) const override;
|
| D | xla_device_context.cc | 117 bool sync_dst_compute) const { in CopyCPUTensorToDevice()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 80 bool sync_dst_compute = true) const {
|