Searched refs:out_allocator (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 56 Allocator* out_allocator, StringPiece edge_name, in CopyHostToDevice() argument
69 auto copier = [dst, recv_dev_context, out_allocator, status_cb, in CopyHostToDevice()
75 CopyHostToDevice(&from, cpu_allocator, out_allocator, edge_name, dst, in CopyHostToDevice()
89 *to = Tensor(out_allocator, from.dtype(), from.shape()); in CopyHostToDevice()
123 Allocator* cpu_allocator, Allocator* out_allocator, in CopyDeviceToDevice() argument
141 out_allocator, status_cb, dev_to_dev_stream_index, in CopyDeviceToDevice()
147 CopyDeviceToDevice(copy_function, cpu_allocator, out_allocator, in CopyDeviceToDevice()
163 *to = Tensor(out_allocator, from.dtype(), from.shape()); in CopyDeviceToDevice()
226 Allocator* out_allocator = dst->GetAllocator(dst_alloc_attr); in ViaDMA() local
241 CopyDeviceToDevice(ri.copy_function, cpu_allocator, out_allocator, in ViaDMA()
[all …]
|
| D | copy_tensor.h | 72 Allocator* out_allocator, StringPiece edge_name,
|
| D | rendezvous_mgr.cc | 97 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone() local
116 Tensor copy(out_allocator, in.dtype(), in.shape(), aa); in SameWorkerRecvDone()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 281 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone() local
294 Tensor copy(out_allocator, in.dtype(), in.shape(), allocation_attr); in SameWorkerRecvDone()
|