Searched refs:dst_dev_ctx (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 84 DeviceContext* src_dev_ctx, DeviceContext* dst_dev_ctx, Device* src_dev, in MemCpyAsync() argument
109 if (dst_dev_ctx == nullptr && dst_device_type == DEVICE_GPU) { in MemCpyAsync()
113 dst_dev_ctx = dev_info->default_context; in MemCpyAsync()
116 if (non_cpu_dst) CHECK(dst_dev_ctx); in MemCpyAsync()
119 src_dev_ctx, dst_dev_ctx, src_dev, dst_dev, src_attr, in MemCpyAsync()
|
| D | collective_rma_local.h | 77 DeviceContext* dst_dev_ctx, Device* src_dev,
|