Searched refs:out_allocator (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 51 Allocator* out_allocator, StringPiece edge_name, in CopyHostToDevice() argument
64 [dst, recv_dev_context, out_allocator, status_cb]( in CopyHostToDevice()
78 *to = Tensor(out_allocator, from.dtype(), from.shape()); in CopyHostToDevice()
109 Allocator* out_allocator, StringPiece edge_name, in CopyDeviceToHost() argument
122 [edge_name, src, send_dev_context, out_allocator, status_cb]( in CopyDeviceToHost()
136 *to = Tensor(out_allocator, from.dtype(), from.shape()); in CopyDeviceToHost()
167 Allocator* cpu_allocator, Allocator* out_allocator, in CopyDeviceToDevice() argument
185 recv_dev_context, send_dev_context, out_allocator, in CopyDeviceToDevice()
199 *to = Tensor(out_allocator, from.dtype(), from.shape()); in CopyDeviceToDevice()
256 Allocator* out_allocator = dst->GetAllocator(dst_alloc_attr); in ViaDMA() local
[all …]
|
| D | rendezvous_mgr.cc | 103 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/ |
| D | base_rendezvous_mgr.cc | 271 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone() local
275 Tensor copy(out_allocator, in.dtype(), in.shape()); in SameWorkerRecvDone()
|