Searched defs:src_alloc_attr (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 126 Device* dst, const AllocatorAttributes src_alloc_attr, in CopyDeviceToDevice()
204 Device* dst, const AllocatorAttributes src_alloc_attr, in ViaDMA()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_util.cc | 112 Device* src, Device* dst, AllocatorAttributes src_alloc_attr, in DeviceToDeviceCopy()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 189 Device* src, Device* dst, AllocatorAttributes src_alloc_attr, in DeviceToDeviceCopy()
|