Home
last modified time | relevance | path

Searched refs:recv_dev_context (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc58 DeviceContext* recv_dev_context, StatusCallback done, in CopyHostToDevice() argument
69 auto copier = [dst, recv_dev_context, out_allocator, status_cb, in CopyHostToDevice()
76 to, recv_dev_context, wrapped_done, sync_dst_compute); in CopyHostToDevice()
90 recv_dev_context->CopyCPUTensorToDevice(&from, dst, to, wrapped_done, in CopyHostToDevice()
117 recv_dev_context->CopyCPUTensorToDevice(input, dst, output, std::move(done), in CopyHostToDevice()
125 DeviceContext* recv_dev_context, Device* src, in CopyDeviceToDevice() argument
140 dst_alloc_attr, recv_dev_context, send_dev_context, in CopyDeviceToDevice()
148 send_dev_context, recv_dev_context, src, dst, in CopyDeviceToDevice()
164 copy_function(send_dev_context, recv_dev_context, src, dst, in CopyDeviceToDevice()
193 copy_function(send_dev_context, recv_dev_context, src, dst, src_alloc_attr, in CopyDeviceToDevice()
[all …]
Dcopy_tensor.h32 DeviceContext* send_dev_context, DeviceContext* recv_dev_context,
44 DeviceContext* recv_dev_context, Device* src, Device* dst,
/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device_util.h55 DeviceContext* send_dev_context, DeviceContext* recv_dev_context,
Dpluggable_device_util.cc111 DeviceContext* send_dev_context, DeviceContext* recv_dev_context, in DeviceToDeviceCopy() argument
143 static_cast<const PluggableDeviceContext*>(recv_dev_context)->stream(); in DeviceToDeviceCopy()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.h94 DeviceContext* send_dev_context, DeviceContext* recv_dev_context,
Dgpu_util.cc188 DeviceContext* send_dev_context, DeviceContext* recv_dev_context, in DeviceToDeviceCopy() argument
218 static_cast<const GPUDeviceContext*>(recv_dev_context)->stream(); in DeviceToDeviceCopy()