Searched refs:src_alloc_attr (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | copy_tensor.cc | 126 Device* dst, const AllocatorAttributes src_alloc_attr, in CopyDeviceToDevice() argument
139 auto copier = [copy_function, cpu_allocator, src, dst, src_alloc_attr, in CopyDeviceToDevice()
149 src_alloc_attr, dst_alloc_attr, &from, to, in CopyDeviceToDevice()
165 src_alloc_attr, dst_alloc_attr, &from, to, in CopyDeviceToDevice()
193 copy_function(send_dev_context, recv_dev_context, src, dst, src_alloc_attr, in CopyDeviceToDevice()
204 Device* dst, const AllocatorAttributes src_alloc_attr, in ViaDMA() argument
214 src_alloc_attr.on_host() ? DEVICE_CPU : src->attributes().device_type()); in ViaDMA()
243 src_alloc_attr, dst_alloc_attr, input, output, in ViaDMA()
|
| H A D | copy_tensor.h | 33 Device* src, Device* dst, const AllocatorAttributes src_alloc_attr,
45 const AllocatorAttributes src_alloc_attr,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_util.h | 56 Device* src, Device* dst, AllocatorAttributes src_alloc_attr,
|
| H A D | pluggable_device_util.cc | 112 Device* src, Device* dst, AllocatorAttributes src_alloc_attr, in DeviceToDeviceCopy() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_util.h | 95 Device* src, Device* dst, AllocatorAttributes src_alloc_attr,
|
| H A D | gpu_util.cc | 189 Device* src, Device* dst, AllocatorAttributes src_alloc_attr, in DeviceToDeviceCopy() argument
|