Searched defs:GetScopedAllocator (Results 1 – 4 of 4) sorted by relevance
80 Allocator* ThreadPoolDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function in tensorflow::ThreadPoolDevice
62 Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function
196 virtual Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function
964 Allocator* BaseGPUDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function in tensorflow::BaseGPUDevice