Searched refs:sync_dst_compute (Results 1 – 11 of 11) 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
257 sync_dst_compute](Status status) { in ViaDMA()
264 std::move(delete_and_done), sync_dst_compute); in ViaDMA()
285 sync_dst_compute); in ViaDMA()
|
| D | rendezvous_mgr.cc | 82 bool sync_dst_compute = true; in SameWorkerRecvDone() local
96 sync_dst_compute = false; in SameWorkerRecvDone()
105 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/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 285 bool sync_dst_compute = (safe_alloc_frontier == 0); in SameWorkerRecvDone() local
290 if (!sync_dst_compute) { in SameWorkerRecvDone()
304 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 | 115 bool sync_dst_compute) const { in CopyCPUTensorToDevice()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 83 bool sync_dst_compute = true) const {
|