Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcopy_tensor.cc127 const AllocatorAttributes dst_alloc_attr, in CopyDeviceToDevice()
205 const AllocatorAttributes dst_alloc_attr, in ViaDMA()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_util.cc113 AllocatorAttributes dst_alloc_attr, const Tensor* input, Tensor* output, in DeviceToDeviceCopy()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_util.cc190 AllocatorAttributes dst_alloc_attr, const Tensor* input, Tensor* output, in DeviceToDeviceCopy()