Searched refs:src_alloc_attr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 124 Device* dst, const AllocatorAttributes src_alloc_attr, in CopyDeviceToDevice() argument
138 [copy_function, cpu_allocator, src, dst, src_alloc_attr, dst_alloc_attr, in CopyDeviceToDevice()
147 src_alloc_attr, dst_alloc_attr, &from, to, in CopyDeviceToDevice()
163 src_alloc_attr, dst_alloc_attr, &from, to, in CopyDeviceToDevice()
191 copy_function(send_dev_context, recv_dev_context, src, dst, src_alloc_attr, in CopyDeviceToDevice()
202 Device* dst, const AllocatorAttributes src_alloc_attr, in ViaDMA() argument
212 src_alloc_attr.on_host() ? DEVICE_CPU : src->attributes().device_type()); in ViaDMA()
236 src_alloc_attr, dst_alloc_attr, input, output, in ViaDMA()
|
| D | copy_tensor.h | 33 Device* src, Device* dst, const AllocatorAttributes src_alloc_attr,
45 const AllocatorAttributes src_alloc_attr,
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.h | 95 Device* src, Device* dst, AllocatorAttributes src_alloc_attr,
|
| D | gpu_util.cc | 189 Device* src, Device* dst, AllocatorAttributes src_alloc_attr, in DeviceToDeviceCopy() argument
|