Home
last modified time | relevance | path

Searched defs:src_alloc_attr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc124 Device* dst, const AllocatorAttributes src_alloc_attr, in CopyDeviceToDevice()
202 Device* dst, const AllocatorAttributes src_alloc_attr, in ViaDMA()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc189 Device* src, Device* dst, AllocatorAttributes src_alloc_attr, in DeviceToDeviceCopy()