Home
last modified time | relevance | path

Searched refs:non_cpu_dst (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local.cc134 const bool non_cpu_dst = dst_device_type != DeviceType(DEVICE_CPU); in MemCpyAsync() local
151 if (non_cpu_dst) CHECK(dst_dev_ctx); in MemCpyAsync()
152 if (non_cpu_src || non_cpu_dst) { in MemCpyAsync()
Dcopy_tensor.cc216 const bool non_cpu_dst = dst_device_type != DeviceType(DEVICE_CPU); in ViaDMA() local
227 if (non_cpu_src && non_cpu_dst) { in ViaDMA()
274 if (non_cpu_src && !non_cpu_dst) { in ViaDMA()
282 if (!non_cpu_src && non_cpu_dst) { in ViaDMA()
291 CHECK(!non_cpu_src && !non_cpu_dst); in ViaDMA()