Searched refs:recv_dev_context (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 54 DeviceContext* recv_dev_context, StatusCallback done, in CopyHostToDevice() argument
66 [dst, recv_dev_context, out_allocator, status_cb, cpu_allocator, in CopyHostToDevice()
72 dst, to, recv_dev_context, wrapped_done, in CopyHostToDevice()
87 recv_dev_context->CopyCPUTensorToDevice( in CopyHostToDevice()
114 recv_dev_context->CopyCPUTensorToDevice(input, dst, output, std::move(done), in CopyHostToDevice()
123 DeviceContext* recv_dev_context, Device* src, in CopyDeviceToDevice() argument
139 recv_dev_context, send_dev_context, out_allocator, status_cb, in CopyDeviceToDevice()
146 send_dev_context, recv_dev_context, src, dst, in CopyDeviceToDevice()
162 copy_function(send_dev_context, recv_dev_context, src, dst, in CopyDeviceToDevice()
191 copy_function(send_dev_context, recv_dev_context, src, dst, src_alloc_attr, in CopyDeviceToDevice()
[all …]
|
| D | copy_tensor.h | 32 DeviceContext* send_dev_context, DeviceContext* recv_dev_context,
44 DeviceContext* recv_dev_context, Device* src, Device* dst,
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.h | 94 DeviceContext* send_dev_context, DeviceContext* recv_dev_context,
|
| D | gpu_util.cc | 188 DeviceContext* send_dev_context, DeviceContext* recv_dev_context, in DeviceToDeviceCopy() argument
218 static_cast<const GPUDeviceContext*>(recv_dev_context)->stream(); in DeviceToDeviceCopy()
|