Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc125 const AllocatorAttributes dst_alloc_attr, in CopyDeviceToDevice()
203 const AllocatorAttributes dst_alloc_attr, in ViaDMA()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc190 AllocatorAttributes dst_alloc_attr, const Tensor* input, Tensor* output, in DeviceToDeviceCopy()