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.cc98 const bool non_cpu_src = src_device_type != DeviceType(DEVICE_CPU); in MemCpyAsync() local
115 if (non_cpu_src) CHECK(src_dev_ctx); in MemCpyAsync()
117 if (non_cpu_src || non_cpu_dst) { in MemCpyAsync()
Dcopy_tensor.cc262 const bool non_cpu_src = src_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()