Searched defs:out_allocator (Results 1 – 3 of 3) sorted by relevance
51 Allocator* out_allocator, StringPiece edge_name, in CopyHostToDevice()116 Allocator* out_allocator, StringPiece edge_name, in CopyDeviceToHost()181 Allocator* cpu_allocator, Allocator* out_allocator, in CopyDeviceToDevice()271 Allocator* out_allocator = dst->GetAllocator(dst_alloc_attr); in ViaDMA() local
103 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone() local
271 Allocator* out_allocator = dst_device->GetAllocator(attr); in SameWorkerRecvDone() local