Searched defs:recv_dev_context (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | copy_tensor.cc | 58 DeviceContext* recv_dev_context, StatusCallback done, in CopyHostToDevice()
125 DeviceContext* recv_dev_context, Device* src, in CopyDeviceToDevice()
203 DeviceContext* recv_dev_context, Device* src, in ViaDMA()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_util.cc | 111 DeviceContext* send_dev_context, DeviceContext* recv_dev_context, in DeviceToDeviceCopy()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_util.cc | 188 DeviceContext* send_dev_context, DeviceContext* recv_dev_context, in DeviceToDeviceCopy()
|