Home
last modified time | relevance | path

Searched refs:out_allocator (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc51 Allocator* out_allocator, StringPiece edge_name, in CopyHostToDevice() argument
64 [dst, recv_dev_context, out_allocator, status_cb, cpu_allocator, in CopyHostToDevice()
70 CopyHostToDevice(&from, cpu_allocator, out_allocator, edge_name, in CopyHostToDevice()
84 *to = Tensor(out_allocator, from.dtype(), from.shape()); in CopyHostToDevice()
116 Allocator* out_allocator, StringPiece edge_name, in CopyDeviceToHost() argument
129 [edge_name, src, send_dev_context, out_allocator, status_cb, in CopyDeviceToHost()
135 CopyDeviceToHost(&from, cpu_allocator, out_allocator, edge_name, in CopyDeviceToHost()
149 *to = Tensor(out_allocator, from.dtype(), from.shape()); in CopyDeviceToHost()
181 Allocator* cpu_allocator, Allocator* out_allocator, in CopyDeviceToDevice() argument
199 recv_dev_context, send_dev_context, out_allocator, status_cb, in CopyDeviceToDevice()
[all …]
Drendezvous_mgr.cc103 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone() local
106 Tensor copy(out_allocator, in.dtype(), in.shape()); in SameWorkerRecvDone()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dbase_rendezvous_mgr.cc271 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone() local
275 Tensor copy(out_allocator, in.dtype(), in.shape()); in SameWorkerRecvDone()