Searched refs:recv_dev_context (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 53 DeviceContext* recv_dev_context, StatusCallback done) { in CopyHostToDevice() argument
64 [dst, recv_dev_context, out_allocator, status_cb]( in CopyHostToDevice()
79 recv_dev_context->CopyCPUTensorToDevice(&from, dst, to, in CopyHostToDevice()
103 recv_dev_context->CopyCPUTensorToDevice(input, dst, output, in CopyHostToDevice()
169 DeviceContext* recv_dev_context, Device* src, in CopyDeviceToDevice() argument
185 recv_dev_context, send_dev_context, out_allocator, in CopyDeviceToDevice()
200 copy_function(send_dev_context, recv_dev_context, src, dst, in CopyDeviceToDevice()
226 copy_function(send_dev_context, recv_dev_context, src, dst, src_alloc_attr, in CopyDeviceToDevice()
235 DeviceContext* recv_dev_context, Device* src, in ViaDMA() argument
267 send_dev_context, recv_dev_context, src, dst, in ViaDMA()
[all …]
|
| D | copy_tensor.h | 32 DeviceContext* recv_dev_context, Device* src,
46 DeviceContext* recv_dev_context, Device* src, Device* dst,
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.h | 97 DeviceContext* recv_dev_context, Device* src,
|
| D | gpu_util.cc | 191 DeviceContext* recv_dev_context, Device* src, in DeviceToDeviceCopy() argument
223 static_cast<const GPUDeviceContext*>(recv_dev_context)->stream(); in DeviceToDeviceCopy()
|