send_device_to_device_stream (reference) in projects: aosp_15_r20

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:send_device_to_device_stream (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A D	pluggable_device_util.cc	124 auto send_device_to_device_stream = in DeviceToDeviceCopy() local 127 if (send_device_to_device_stream == nullptr) { in DeviceToDeviceCopy() 134 send_device_to_device_stream->ThenWaitFor(send_stream); in DeviceToDeviceCopy() 151 send_device_to_device_stream->ThenWaitFor(recv_stream); in DeviceToDeviceCopy() 154 send_device_to_device_stream->ThenMemcpy(&device_dst_ptr, device_src_ptr, in DeviceToDeviceCopy() 160 send_device_to_device_stream, in DeviceToDeviceCopy() 161 [done, send_device_to_device_stream, input_ref]() { in DeviceToDeviceCopy() 163 if (!send_device_to_device_stream->ok()) { in DeviceToDeviceCopy() 170 send_device_to_device_stream); in DeviceToDeviceCopy()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A D	gpu_util.cc	200 auto send_device_to_device_stream = in DeviceToDeviceCopy() local 203 if (send_device_to_device_stream == nullptr) { in DeviceToDeviceCopy() 209 send_device_to_device_stream->ThenWaitFor(send_stream); in DeviceToDeviceCopy() 228 send_device_to_device_stream->ThenWaitFor(recv_stream); in DeviceToDeviceCopy() 231 send_device_to_device_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, in DeviceToDeviceCopy() 238 send_device_to_device_stream, in DeviceToDeviceCopy() 239 [done, send_device_to_device_stream, input_ref]() { in DeviceToDeviceCopy() 241 if (!send_device_to_device_stream->ok()) { in DeviceToDeviceCopy() 247 send_device_to_device_stream); in DeviceToDeviceCopy()