Searched defs:out_allocator (Results 1 – 3 of 3) sorted by relevance
52 Allocator* out_allocator, StringPiece edge_name, in CopyHostToDevice()121 Allocator* cpu_allocator, Allocator* out_allocator, in CopyDeviceToDevice()223 Allocator* out_allocator = dst->GetAllocator(dst_alloc_attr); in ViaDMA() local332 Allocator* out_allocator, StringPiece edge_name, in CopyDeviceToHost()
81 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone() local
282 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone() local