Searched refs:dst_alloc_attr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 185 const AllocatorAttributes dst_alloc_attr, in CopyDeviceToDevice() argument
198 [copy_function, src, dst, src_alloc_attr, dst_alloc_attr, in CopyDeviceToDevice()
215 src_alloc_attr, dst_alloc_attr, &from, to, in CopyDeviceToDevice()
241 dst_alloc_attr, input, output, dev_to_dev_stream_index, in CopyDeviceToDevice()
252 const AllocatorAttributes dst_alloc_attr, in ViaDMA() argument
261 dst_alloc_attr.on_host() ? DEVICE_CPU : dst->attributes().device_type()); in ViaDMA()
271 Allocator* out_allocator = dst->GetAllocator(dst_alloc_attr); in ViaDMA()
283 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 | 191 AllocatorAttributes dst_alloc_attr, const Tensor* input, Tensor* output, in DeviceToDeviceCopy() argument
|