Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local.cc133 const bool non_cpu_src = src_device_type != DeviceType(DEVICE_CPU); in MemCpyAsync() local
150 if (non_cpu_src) CHECK(src_dev_ctx); in MemCpyAsync()
152 if (non_cpu_src || non_cpu_dst) { in MemCpyAsync()
Dcopy_tensor.cc215 const bool non_cpu_src = src_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()