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.cc99 const bool non_cpu_dst = dst_device_type != DeviceType(DEVICE_CPU); in MemCpyAsync() local
116 if (non_cpu_dst) CHECK(dst_dev_ctx); in MemCpyAsync()
117 if (non_cpu_src || non_cpu_dst) { in MemCpyAsync()
Dcopy_tensor.cc263 const bool non_cpu_dst = dst_device_type != DeviceType(DEVICE_CPU); in ViaDMA() local
274 if (non_cpu_src && non_cpu_dst) { in ViaDMA()
326 if (non_cpu_src && !non_cpu_dst) { in ViaDMA()
334 if (!non_cpu_src && non_cpu_dst) { in ViaDMA()
342 CHECK(!non_cpu_src && !non_cpu_dst); in ViaDMA()