Searched defs:out_allocator (Results 1 – 3 of 3) sorted by relevance
56 Allocator* out_allocator, StringPiece edge_name, in CopyHostToDevice()123 Allocator* cpu_allocator, Allocator* out_allocator, in CopyDeviceToDevice()226 Allocator* out_allocator = dst->GetAllocator(dst_alloc_attr); in ViaDMA() local341 Allocator* out_allocator, StringPiece edge_name, in CopyDeviceToHost()
97 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone() local
281 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone() local