Searched refs:dst_alloc_attr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 125 const AllocatorAttributes dst_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()
192 dst_alloc_attr, input, output, dev_to_dev_stream_index, in CopyDeviceToDevice()
203 const AllocatorAttributes dst_alloc_attr, in ViaDMA() argument
214 dst_alloc_attr.on_host() ? DEVICE_CPU : dst->attributes().device_type()); in ViaDMA()
224 Allocator* out_allocator = dst->GetAllocator(dst_alloc_attr); in ViaDMA()
236 src_alloc_attr, dst_alloc_attr, input, output, in ViaDMA()
|
| D | copy_tensor.h | 34 const AllocatorAttributes dst_alloc_attr, const Tensor* input,
46 const AllocatorAttributes dst_alloc_attr,
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.h | 96 AllocatorAttributes dst_alloc_attr, const Tensor* input, Tensor* output,
|
| D | gpu_util.cc | 190 AllocatorAttributes dst_alloc_attr, const Tensor* input, Tensor* output, in DeviceToDeviceCopy() argument
|